/[thuban]/branches/WIP-pyshapelib-bramz/test/test_transientdb.py
ViewVC logotype

Diff of /branches/WIP-pyshapelib-bramz/test/test_transientdb.py

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 805 by jan, Fri May 2 16:42:35 2003 UTC revision 845 by bh, Tue May 6 18:05:13 2003 UTC
# Line 20  import support Line 20  import support
20  support.initthuban()  support.initthuban()
21    
22  from Thuban.Model.table import DBFTable, MemoryTable, FIELDTYPE_STRING, \  from Thuban.Model.table import DBFTable, MemoryTable, FIELDTYPE_STRING, \
23                                 FIELDTYPE_INT                                 FIELDTYPE_INT, FIELDTYPE_DOUBLE
24  from Thuban.Model.transientdb import TransientDatabase, TransientTable, \  from Thuban.Model.transientdb import TransientDatabase, TransientTable, \
25       TransientJoinedTable, AutoTransientTable       TransientJoinedTable, AutoTransientTable
26    
# Line 47  class TestTransientTable(unittest.TestCa Line 47  class TestTransientTable(unittest.TestCa
47          Assume that table holds the data of the file          Assume that table holds the data of the file
48          ../Data/iceland/political.dbf sample file.          ../Data/iceland/political.dbf sample file.
49          """          """
50          self.assertEquals(table.record_count(), 156)          self.assertEquals(table.NumRows(), 156)
51          self.assertEquals(table.field_count(), 8)          self.assertEquals(table.NumColumns(), 8)
52    
53          # Check one each of the possible field types. The width and          # Check one each of the possible field types. The width and
54          # decimal precision is always 0.          # decimal precision is always 0.
55          self.assertEquals(table.field_info(0), ('double', 'AREA', 0, 0))          columns = table.Columns()
56          self.assertEquals(table.field_info(3), ('int', 'PONET_ID', 0, 0))          self.assertEquals(columns[0].name, 'AREA')
57          self.assertEquals(table.field_info(6), ('string', 'POPYCOUN', 0, 0))          self.assertEquals(columns[0].type, FIELDTYPE_DOUBLE)
58            self.assertEquals(columns[3].name, 'PONET_ID')
59            self.assertEquals(columns[3].type, FIELDTYPE_INT)
60            self.assertEquals(columns[6].name, 'POPYCOUN')
61            self.assertEquals(columns[6].type, FIELDTYPE_STRING)
62    
63            # HasColumn
64            self.failUnless(table.HasColumn("AREA"))
65            self.failUnless(table.HasColumn(1))
66            # HasColumn for non-exisiting columns
67            self.failIf(table.HasColumn("non_existing_name"))
68            self.failIf(table.HasColumn(100))
69    
70          # Read an `interesting' record          # Read an `interesting' record
71          self.assertEquals(table.read_record(144),          self.assertEquals(table.ReadRowAsDict(144),
72                            {'POPYCOUN': 'IC', 'POPYADMIN': '', 'PONET_': 146,                            {'POPYCOUN': 'IC', 'POPYADMIN': '', 'PONET_': 146,
73                             'AREA': 19.462,                             'AREA': 19.462,
74                             'POPYTYPE': 1, 'PERIMETER': 88.518000000000001,                             'POPYTYPE': 1, 'PERIMETER': 88.518000000000001,
75                             'POPYREG': '1',                             'POPYREG': '1',
76                             'PONET_ID': 145})                             'PONET_ID': 145})
77    
78          # field_range may induce a copy to the transient database.          # ValueRange may induce a copy to the transient database.
79          # Therefore we put it last so that we can execute this method          # Therefore we put it last so that we can execute this method
80          # twice to check whether the other methods still work after the          # twice to check whether the other methods still work after the
81          # copy          # copy
82          self.assertEquals(table.field_range("AREA"),          self.assertEquals(table.ValueRange("AREA"), (0.0, 19.462))
                           ((0.0, None), (19.462, None)))  
83    
84          unique = table.GetUniqueValues("PONET_ID")          unique = table.UniqueValues("PONET_ID")
85          unique.sort()          unique.sort()
86          self.assertEquals(unique, range(1, 157))          self.assertEquals(unique, range(1, 157))
87    
# Line 108  class TestTransientTable(unittest.TestCa Line 118  class TestTransientTable(unittest.TestCa
118          self.run_iceland_political_tests(table)          self.run_iceland_political_tests(table)
119          self.run_iceland_political_tests(table)          self.run_iceland_political_tests(table)
120    
121        def test_auto_transient_table_query(self):
122            """Test AutoTransientTable.SimpleQuery()"""
123            orig_table = DBFTable(os.path.join("..", "Data", "iceland",
124                                               "political.dbf"))
125            table = AutoTransientTable(self.transientdb, orig_table)
126            # Only a simple test here. The AutoTransientTable siply
127            # delegates to its transient table so it should be OK that the
128            # real test for it is in test_transient_table_query. However,
129            # it's important to check that the column handling works
130            # correctly because the AutoTransientTable and it's underlying
131            # transient table use different column object types.
132            self.assertEquals(table.SimpleQuery(table.Column("AREA"), ">", 10.0),
133                              [144])
134    
135      def test_transient_joined_table(self):      def test_transient_joined_table(self):
136          """Test TransientJoinedTable"""          """Test TransientJoinedTable"""
# Line 127  class TestTransientTable(unittest.TestCa Line 150  class TestTransientTable(unittest.TestCa
150          table = TransientJoinedTable(self.transientdb, landmarks, "CLPTLABEL",          table = TransientJoinedTable(self.transientdb, landmarks, "CLPTLABEL",
151                                       auto, "type")                                       auto, "type")
152    
153          self.assertEquals(table.record_count(), 34)          self.assertEquals(table.NumRows(), 34)
154          self.assertEquals(table.field_count(), 8)          self.assertEquals(table.NumColumns(), 8)
155          self.assertEquals(table.field_info(0), ('double', 'AREA', 0, 0))          self.assertEquals(table.Column(0).type, FIELDTYPE_DOUBLE)
156          self.assertEquals(table.field_info(7), ('int', 'code', 0, 0))          self.assertEquals(table.Column(0).name, 'AREA')
157          self.assertEquals(table.field_info(4), ('string', 'CLPTLABEL', 0, 0))          self.assertEquals(table.Column(7).type, FIELDTYPE_INT)
158            self.assertEquals(table.Column(7).name, 'code')
159            self.assertEquals(table.Column(4).type, FIELDTYPE_STRING)
160            self.assertEquals(table.Column(4).name, 'CLPTLABEL')
161            # HasColumn
162            self.failUnless(table.HasColumn("AREA"))
163            self.failUnless(table.HasColumn(1))
164            # HasColumn for non-exisiting columns
165            self.failIf(table.HasColumn("non_existing_name"))
166            self.failIf(table.HasColumn(100))
167    
168          # Read an `interesting' record          # Read an `interesting' record
169          self.assertEquals(table.read_record(22),          self.assertEquals(table.ReadRowAsDict(22),
170                            {'PERIMETER': 0.0, 'CLPOINT_': 23,                            {'PERIMETER': 0.0, 'CLPOINT_': 23,
171                             'AREA': 0.0, 'CLPTLABEL': 'RUINS',                             'AREA': 0.0, 'CLPTLABEL': 'RUINS',
172                             'CLPOINT_ID': 38, 'CLPTFLAG': 0,                             'CLPOINT_ID': 38, 'CLPTFLAG': 0,
# Line 145  class TestTransientTable(unittest.TestCa Line 177  class TestTransientTable(unittest.TestCa
177    
178    
179      def test_transient_table_read_twice(self):      def test_transient_table_read_twice(self):
180          """Test TransientTable.read_record() reading the same record twice"""          """Test TransientTable.ReadRowAsDict() reading the same record twice"""
181          simple = MemoryTable([("type", FIELDTYPE_STRING),          simple = MemoryTable([("type", FIELDTYPE_STRING),
182                                ("code", FIELDTYPE_INT)],                                ("code", FIELDTYPE_INT)],
183                               [("OTHER/UNKNOWN", 0),                               [("OTHER/UNKNOWN", 0),
# Line 161  class TestTransientTable(unittest.TestCa Line 193  class TestTransientTable(unittest.TestCa
193          # unitialized local variable, so for passing the test it's          # unitialized local variable, so for passing the test it's
194          # enough if reading simply succeeds. OTOH, while we're at it we          # enough if reading simply succeeds. OTOH, while we're at it we
195          # might as well check whether the results are equal anyway :)          # might as well check whether the results are equal anyway :)
196          result1 = table.read_record(3)          result1 = table.ReadRowAsDict(3)
197          result2 = table.read_record(3)          result2 = table.ReadRowAsDict(3)
198          self.assertEquals(result1, result2)          self.assertEquals(result1, result2)
199    
200    
201        def test_transient_table_query(self):
202            """Test TransientTable.SimpleQuery()"""
203            simple = MemoryTable([("type", FIELDTYPE_STRING),
204                                  ("value", FIELDTYPE_DOUBLE),
205                                  ("code", FIELDTYPE_INT)],
206                                 [("OTHER/UNKNOWN", -1.5, 11),
207                                  ("RUINS", 0.0, 1),
208                                  ("FARM", 3.141, 2),
209                                  ("BUILDING", 2.5, 3),
210                                  ("HUT", 1e6, 4),
211                                  ("LIGHTHOUSE", -0.01, 5)])
212            table = TransientTable(self.transientdb, simple)
213    
214            # A column and a value
215            self.assertEquals(table.SimpleQuery(table.Column(0), "==", "RUINS"),
216                              [1])
217            self.assertEquals(table.SimpleQuery(table.Column(2), "!=", 2),
218                              [0, 1, 3, 4, 5])
219            self.assertEquals(table.SimpleQuery(table.Column(1), "<", 1.0),
220                              [0, 1, 5])
221            self.assertEquals(table.SimpleQuery(table.Column(1), "<=", -1.5),
222                              [0])
223            self.assertEquals(table.SimpleQuery(table.Column(2), ">", 3),
224                              [0, 4, 5])
225            self.assertEquals(table.SimpleQuery(table.Column(2), ">=", 3),
226                              [0, 3, 4, 5])
227    
228            # Two columns as operands
229            self.assertEquals(table.SimpleQuery(table.Column(1),
230                                                "<=", table.Column(2)),
231                              [0, 1, 3, 5])
232    
233            # Test whether invalid operators raise a ValueError
234            self.assertRaises(ValueError,
235                              table.SimpleQuery,
236                              table.Column(1), "<<", table.Column(2))
237    
238    
239  if __name__ == "__main__":  if __name__ == "__main__":
240      support.run_tests()      support.run_tests()
Legend:
Removed from v.805  
changed lines
  Added in v.845
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26