/[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 765 by bh, Tue Apr 29 12:42:14 2003 UTC revision 998 by bh, Thu May 22 19:29:39 2003 UTC
# Line 19  import unittest Line 19  import unittest
19  import support  import support
20  support.initthuban()  support.initthuban()
21    
22  from Thuban.Model.table import DBFTable, FIELDTYPE_STRING, FIELDTYPE_INT  from Thuban.Model.table import DBFTable, MemoryTable, FIELDTYPE_STRING, \
23                                   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    
27    
 class SimpleTable:  
   
     """Very simple table implementation that operates on a list of tuples"""  
   
     def __init__(self, fields, data):  
         """Initialize the SimpleTable  
   
         Parameters:  
         fields -- List of (name, field_type) pairs  
         data -- List of tuples, one for each row of data  
         """  
         self.fields = fields  
         self.data = data  
   
     def field_count(self):  
         return len(self.fields)  
   
     def field_info(self, index):  
         name, type = self.fields[index]  
         return (type, name, 0, 0)  
   
     def record_count(self):  
         return len(self.data)  
   
     def read_record(self, index):  
         return dict([(self.fields[i][0], self.data[index][i])  
                       for i in range(len(self.fields))])  
   
   
28  class TestTransientTable(unittest.TestCase, support.FileTestMixin):  class TestTransientTable(unittest.TestCase, support.FileTestMixin):
29    
30      def setUp(self):      def setUp(self):
# Line 75  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          # Reading rows and values.
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            self.assertEquals(table.ReadValue(144, "AREA"), 19.462)
78            self.assertEquals(table.ReadValue(144, 3), 145)
79    
80          # field_range may induce a copy to the transient database.          # ValueRange may induce a copy to the transient database.
81          # Therefore we put it last so that we can execute this method          # Therefore we put it last so that we can execute this method
82          # twice to check whether the other methods still work after the          # twice to check whether the other methods still work after the
83          # copy          # copy
84          self.assertEquals(table.field_range("AREA"),          self.assertEquals(table.ValueRange("AREA"), (0.0, 19.462))
                           ((0.0, None), (19.462, None)))  
85    
86          unique = table.GetUniqueValues("PONET_ID")          unique = table.UniqueValues("PONET_ID")
87          unique.sort()          unique.sort()
88          self.assertEquals(unique, range(1, 157))          self.assertEquals(unique, range(1, 157))
89    
# Line 117  class TestTransientTable(unittest.TestCa Line 101  class TestTransientTable(unittest.TestCa
101          # The transient_table method should return the table itself          # The transient_table method should return the table itself
102          self.assert_(table is table.transient_table())          self.assert_(table is table.transient_table())
103    
104            # The title is simply copied over from the original table
105            self.assertEquals(table.Title(), orig_table.Title())
106    
107            # The TransientTable class itself doesn't implement the
108            # Dependencies method, so we don't test it.
109    
110    
111      def test_auto_transient_table(self):      def test_auto_transient_table(self):
112          """Test AutoTransientTable(dbftable)          """Test AutoTransientTable(dbftable)
# Line 136  class TestTransientTable(unittest.TestCa Line 126  class TestTransientTable(unittest.TestCa
126          self.run_iceland_political_tests(table)          self.run_iceland_political_tests(table)
127          self.run_iceland_political_tests(table)          self.run_iceland_political_tests(table)
128    
129        def test_auto_transient_table_query(self):
130            """Test AutoTransientTable.SimpleQuery()"""
131            orig_table = DBFTable(os.path.join("..", "Data", "iceland",
132                                               "political.dbf"))
133            table = AutoTransientTable(self.transientdb, orig_table)
134            # Only a simple test here. The AutoTransientTable simply
135            # delegates to its transient table so it should be OK that the
136            # real test for it is in test_transient_table_query. However,
137            # it's important to check that the column handling works
138            # correctly because the AutoTransientTable and it's underlying
139            # transient table use different column object types.
140            self.assertEquals(table.SimpleQuery(table.Column("AREA"), ">", 10.0),
141                              [144])
142    
143            # test using a Column object as the right parameter
144            self.assertEquals(table.SimpleQuery(table.Column("POPYTYPE"),
145                                                "==",
146                                                table.Column("POPYREG")),
147                              range(156))
148    
149        def test_auto_transient_table_dependencies(self):
150            """Test AutoTransientTable.Dependencies()"""
151            orig_table = DBFTable(os.path.join("..", "Data", "iceland",
152                                               "political.dbf"))
153            table = AutoTransientTable(self.transientdb, orig_table)
154            self.assertEquals(table.Dependencies(), (orig_table,))
155    
156        def test_auto_transient_table_title(self):
157            """Test AutoTransientTable.Title()"""
158            orig_table = DBFTable(os.path.join("..", "Data", "iceland",
159                                               "political.dbf"))
160            table = AutoTransientTable(self.transientdb, orig_table)
161            # The title is of course the same as that of the original table
162            self.assertEquals(table.Title(), orig_table.Title())
163    
164      def test_transient_joined_table(self):      def test_transient_joined_table(self):
165          """Test TransientJoinedTable"""          """Test TransientJoinedTable"""
166          simple = SimpleTable([("type", FIELDTYPE_STRING),          simple = MemoryTable([("type", FIELDTYPE_STRING),
167                                ("code", FIELDTYPE_INT)],                                ("code", FIELDTYPE_INT)],
168                               [("OTHER/UNKNOWN", 0),                               [("OTHER/UNKNOWN", 0),
169                                ("RUINS", 1),                                ("RUINS", 1),
# Line 155  class TestTransientTable(unittest.TestCa Line 179  class TestTransientTable(unittest.TestCa
179          table = TransientJoinedTable(self.transientdb, landmarks, "CLPTLABEL",          table = TransientJoinedTable(self.transientdb, landmarks, "CLPTLABEL",
180                                       auto, "type")                                       auto, "type")
181    
182          self.assertEquals(table.record_count(), 34)          self.assertEquals(table.NumRows(), 34)
183          self.assertEquals(table.field_count(), 8)          self.assertEquals(table.NumColumns(), 8)
184          self.assertEquals(table.field_info(0), ('double', 'AREA', 0, 0))          self.assertEquals(table.Column(0).type, FIELDTYPE_DOUBLE)
185          self.assertEquals(table.field_info(7), ('int', 'code', 0, 0))          self.assertEquals(table.Column(0).name, 'AREA')
186          self.assertEquals(table.field_info(4), ('string', 'CLPTLABEL', 0, 0))          self.assertEquals(table.Column(7).type, FIELDTYPE_INT)
187            self.assertEquals(table.Column(7).name, 'code')
188            self.assertEquals(table.Column(4).type, FIELDTYPE_STRING)
189            self.assertEquals(table.Column(4).name, 'CLPTLABEL')
190            # HasColumn
191            self.failUnless(table.HasColumn("AREA"))
192            self.failUnless(table.HasColumn(1))
193            # HasColumn for non-exisiting columns
194            self.failIf(table.HasColumn("non_existing_name"))
195            self.failIf(table.HasColumn(100))
196    
197          # Read an `interesting' record          # Reading rows and values
198          self.assertEquals(table.read_record(22),          self.assertEquals(table.ReadRowAsDict(22),
199                            {'PERIMETER': 0.0, 'CLPOINT_': 23,                            {'PERIMETER': 0.0, 'CLPOINT_': 23,
200                             'AREA': 0.0, 'CLPTLABEL': 'RUINS',                             'AREA': 0.0, 'CLPTLABEL': 'RUINS',
201                             'CLPOINT_ID': 38, 'CLPTFLAG': 0,                             'CLPOINT_ID': 38, 'CLPTFLAG': 0,
202                             'code': 1, 'type': 'RUINS'})                             'code': 1, 'type': 'RUINS'})
203            self.assertEquals(table.ReadValue(22, "type"), 'RUINS')
204            self.assertEquals(table.ReadValue(22, 7), 1)
205    
206          # The transient_table method should return the table itself          # The transient_table method should return the table itself
207          self.assert_(table is table.transient_table())          self.assert_(table is table.transient_table())
208    
209            # The TransientJoinedTable depends on both input tables
210            self.assertEquals(table.Dependencies(), (landmarks, auto))
211    
212            # The title is constructed from the titles of the input tables.
213            self.assertEquals(table.Title(),
214                              "Join of %s and %s" % (landmarks.Title(),
215                                                     auto.Title()))
216    
217    
218        def test_transient_table_read_twice(self):
219            """Test TransientTable.ReadRowAsDict() reading the same record twice"""
220            simple = MemoryTable([("type", FIELDTYPE_STRING),
221                                  ("code", FIELDTYPE_INT)],
222                                 [("OTHER/UNKNOWN", 0),
223                                  ("RUINS", 1),
224                                  ("FARM", 2),
225                                  ("BUILDING", 3),
226                                  ("HUT", 4),
227                                  ("LIGHTHOUSE", 5)])
228            table = TransientTable(self.transientdb, simple)
229    
230            # There was a bug where reading the same record twice would
231            # raise an exception in the second call because of an
232            # unitialized local variable, so for passing the test it's
233            # enough if reading simply succeeds. OTOH, while we're at it we
234            # might as well check whether the results are equal anyway :)
235            result1 = table.ReadRowAsDict(3)
236            result2 = table.ReadRowAsDict(3)
237            self.assertEquals(result1, result2)
238    
239    
240        def test_transient_table_query(self):
241            """Test TransientTable.SimpleQuery()"""
242            simple = MemoryTable([("type", FIELDTYPE_STRING),
243                                  ("value", FIELDTYPE_DOUBLE),
244                                  ("code", FIELDTYPE_INT)],
245                                 [("OTHER/UNKNOWN", -1.5, 11),
246                                  ("RUINS", 0.0, 1),
247                                  ("FARM", 3.141, 2),
248                                  ("BUILDING", 2.5, 3),
249                                  ("HUT", 1e6, 4),
250                                  ("LIGHTHOUSE", -0.01, 5)])
251            table = TransientTable(self.transientdb, simple)
252    
253            # A column and a value
254            self.assertEquals(table.SimpleQuery(table.Column(0), "==", "RUINS"),
255                              [1])
256            self.assertEquals(table.SimpleQuery(table.Column(2), "!=", 2),
257                              [0, 1, 3, 4, 5])
258            self.assertEquals(table.SimpleQuery(table.Column(1), "<", 1.0),
259                              [0, 1, 5])
260            self.assertEquals(table.SimpleQuery(table.Column(1), "<=", -1.5),
261                              [0])
262            self.assertEquals(table.SimpleQuery(table.Column(2), ">", 3),
263                              [0, 4, 5])
264            self.assertEquals(table.SimpleQuery(table.Column(2), ">=", 3),
265                              [0, 3, 4, 5])
266    
267            # Two columns as operands
268            self.assertEquals(table.SimpleQuery(table.Column(1),
269                                                "<=", table.Column(2)),
270                              [0, 1, 3, 5])
271    
272            # Test whether invalid operators raise a ValueError
273            self.assertRaises(ValueError,
274                              table.SimpleQuery,
275                              table.Column(1), "<<", table.Column(2))
276    
277    
278  if __name__ == "__main__":  if __name__ == "__main__":
Legend:
Removed from v.765  
changed lines
  Added in v.998
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26