WIP-pyshapelib-bramz/test/test_classification.py

# Copyright (c) 2002 by Intevation GmbH
# Authors:
# Jonathan Coles <[email protected]>
#
# This program is free software under the GPL (>=v2)
# Read the file COPYING coming with Thuban for details.

"""
Test the Classification class
"""

__version__ = "$Revision$"
# $Source$
# $Id$

import unittest

import support
support.initthuban()

from Thuban.Model.classification import Classification
from Thuban.Model.layer import Layer


class TestClassification(unittest.TestCase):

    def test_classification(self):
        """Test Classification"""

        layer = Layer("asdf", "../Data/iceland/political.shp")

        #
        # init with no params
        #
        c = Classification(layer)
        self.assertEqual(c.field, None)
        self.assertNotEqual(c.DefaultData, None)

        #
        # SetField
        #
        c = Classification(layer)
        c.SetField("Test")
        self.assertEqual(c.field, "Test")
        c.SetField(None)
        self.assertEqual(c.field, None)

        #
        # init with field 
        #
        c = Classification(layer, "Test")
        self.assertEqual(c.field, "Test")

        #
        # AddRange
        #
        c0 = Classification(layer, "c0")
        default0 = c0.GetDefaultData()
        self.assertRaises(ValueError, c0.AddRange, 0, 0, None)
        self.assertRaises(ValueError, c0.AddRange, 10, 0, None)
        c0.AddRange(-10, 0, "1")
        c0.AddRange(0, 10, "2")

        #
        # AddPoint
        #
        c1 = Classification(layer, "c1")
        default1 = c1.GetDefaultData()
        c1.AddPoint(0, "1")
        c1.AddPoint("0", "2")
        self.assertRaises(TypeError, c1.AddPoint, {'monty':'python'}, "3")

        #
        # GetProperties
        #

        self.assertEqual(c0.GetProperties(-10), "1")
        self.assertEqual(c0.GetProperties(-11), default0)
        self.assertEqual(c0.GetProperties(0), "2") # min <= x < max
        self.assertEqual(c0.GetProperties(10), default0)
        self.assertEqual(c0.GetProperties(11), default0)

        self.assertEqual(c1.GetProperties(0), "1")
        self.assertEqual(c1.GetProperties("0"), "2")
        self.assertEqual(c1.GetProperties(-1), default1)
        self.assertEqual(c1.GetProperties(1), default1)
        self.assertRaises(TypeError, c1.GetProperties, {'monty':'python'})

        #
        # toggle field
        #

        c0.SetField(None)
        self.assertEqual(c0.GetProperties(10), default0)
        self.assertEqual(c0.GetProperties(11), default0)

        c1.SetField(None)
        self.assertEqual(c1.GetProperties(0), default1)
        self.assertEqual(c1.GetProperties("0"), default1)

        c0.SetField("c0")
        self.assertEqual(c0.GetProperties(10), default0)
        self.assertEqual(c0.GetProperties(11), default0)

        c1.SetField("c1")
        self.assertEqual(c1.GetProperties(0), "1")
        self.assertEqual(c1.GetProperties("0"), "2")

if __name__ == "__main__":
    unittest.main()
1	jonathan	369	# Copyright (c) 2002 by Intevation GmbH
2			# Authors:
3			# Jonathan Coles <[email protected]>
4			#
5			# This program is free software under the GPL (>=v2)
6			# Read the file COPYING coming with Thuban for details.
7
8			"""
9			Test the Classification class
10			"""
11
12			__version__ = "$Revision$"
13			# $Source$
14			# $Id$
15
16			import unittest
17
18			import support
19			support.initthuban()
20
21			from Thuban.Model.classification import Classification
22	jonathan	395	from Thuban.Model.layer import Layer
23	jonathan	369
24
25			class TestClassification(unittest.TestCase):
26
27			def test_classification(self):
28			"""Test Classification"""
29
30	jonathan	395	layer = Layer("asdf", "../Data/iceland/political.shp")
31
32	jonathan	369	#
33			# init with no params
34			#
35	jonathan	395	c = Classification(layer)
36			self.assertEqual(c.field, None)
37			self.assertNotEqual(c.DefaultData, None)
38	jonathan	369
39			#
40	jonathan	395	# SetField
41	jonathan	369	#
42	jonathan	395	c = Classification(layer)
43			c.SetField("Test")
44			self.assertEqual(c.field, "Test")
45			c.SetField(None)
46			self.assertEqual(c.field, None)
47	jonathan	369
48			#
49			# init with field
50			#
51	jonathan	395	c = Classification(layer, "Test")
52			self.assertEqual(c.field, "Test")
53	jonathan	369
54			#
55	jonathan	395	# AddRange
56	jonathan	369	#
57	jonathan	395	c0 = Classification(layer, "c0")
58			default0 = c0.GetDefaultData()
59			self.assertRaises(ValueError, c0.AddRange, 0, 0, None)
60			self.assertRaises(ValueError, c0.AddRange, 10, 0, None)
61			c0.AddRange(-10, 0, "1")
62			c0.AddRange(0, 10, "2")
63	jonathan	369
64			#
65	jonathan	395	# AddPoint
66	jonathan	369	#
67	jonathan	395	c1 = Classification(layer, "c1")
68			default1 = c1.GetDefaultData()
69			c1.AddPoint(0, "1")
70			c1.AddPoint("0", "2")
71			self.assertRaises(TypeError, c1.AddPoint, {'monty':'python'}, "3")
72	jonathan	369
73			#
74	jonathan	395	# GetProperties
75	jonathan	369	#
76
77	jonathan	395	self.assertEqual(c0.GetProperties(-10), "1")
78			self.assertEqual(c0.GetProperties(-11), default0)
79			self.assertEqual(c0.GetProperties(0), "2") # min <= x < max
80			self.assertEqual(c0.GetProperties(10), default0)
81			self.assertEqual(c0.GetProperties(11), default0)
82	jonathan	369
83	jonathan	395	self.assertEqual(c1.GetProperties(0), "1")
84			self.assertEqual(c1.GetProperties("0"), "2")
85			self.assertEqual(c1.GetProperties(-1), default1)
86			self.assertEqual(c1.GetProperties(1), default1)
87			self.assertRaises(TypeError, c1.GetProperties, {'monty':'python'})
88	jonathan	369
89			#
90			# toggle field
91			#
92
93	jonathan	395	c0.SetField(None)
94			self.assertEqual(c0.GetProperties(10), default0)
95			self.assertEqual(c0.GetProperties(11), default0)
96	jonathan	369
97	jonathan	395	c1.SetField(None)
98			self.assertEqual(c1.GetProperties(0), default1)
99			self.assertEqual(c1.GetProperties("0"), default1)
100	jonathan	369
101	jonathan	395	c0.SetField("c0")
102			self.assertEqual(c0.GetProperties(10), default0)
103			self.assertEqual(c0.GetProperties(11), default0)
104	jonathan	369
105	jonathan	395	c1.SetField("c1")
106			self.assertEqual(c1.GetProperties(0), "1")
107			self.assertEqual(c1.GetProperties("0"), "2")
108	jonathan	369
109			if __name__ == "__main__":
110			unittest.main()
Name	Value
svn:eol-style	native
svn:keywords	Author Date Id Revision