50 |
self.assertEquals(table.NumRows(), 156) |
self.assertEquals(table.NumRows(), 156) |
51 |
self.assertEquals(table.NumColumns(), 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. |
|
# decimal precision is always 0. |
|
54 |
columns = table.Columns() |
columns = table.Columns() |
55 |
self.assertEquals(columns[0].name, 'AREA') |
self.assertEquals(columns[0].name, 'AREA') |
56 |
self.assertEquals(columns[0].type, FIELDTYPE_DOUBLE) |
self.assertEquals(columns[0].type, FIELDTYPE_DOUBLE) |
100 |
# The transient_table method should return the table itself |
# The transient_table method should return the table itself |
101 |
self.assert_(table is table.transient_table()) |
self.assert_(table is table.transient_table()) |
102 |
|
|
103 |
|
# The title is simply copied over from the original table |
104 |
|
self.assertEquals(table.Title(), orig_table.Title()) |
105 |
|
|
106 |
# The TransientTable class itself doesn't implement the |
# The TransientTable class itself doesn't implement the |
107 |
# Dependencies method, so we don't test it. |
# Dependencies method, so we don't test it. |
108 |
|
|
152 |
table = AutoTransientTable(self.transientdb, orig_table) |
table = AutoTransientTable(self.transientdb, orig_table) |
153 |
self.assertEquals(table.Dependencies(), (orig_table,)) |
self.assertEquals(table.Dependencies(), (orig_table,)) |
154 |
|
|
155 |
|
def test_auto_transient_table_title(self): |
156 |
|
"""Test AutoTransientTable.Title()""" |
157 |
|
orig_table = DBFTable(os.path.join("..", "Data", "iceland", |
158 |
|
"political.dbf")) |
159 |
|
table = AutoTransientTable(self.transientdb, orig_table) |
160 |
|
# The title is of course the same as that of the original table |
161 |
|
self.assertEquals(table.Title(), orig_table.Title()) |
162 |
|
|
163 |
def test_transient_joined_table(self): |
def test_transient_joined_table(self): |
164 |
"""Test TransientJoinedTable""" |
"""Test TransientJoinedTable""" |
165 |
simple = MemoryTable([("type", FIELDTYPE_STRING), |
simple = MemoryTable([("type", FIELDTYPE_STRING), |
208 |
# The TransientJoinedTable depends on both input tables |
# The TransientJoinedTable depends on both input tables |
209 |
self.assertEquals(table.Dependencies(), (landmarks, auto)) |
self.assertEquals(table.Dependencies(), (landmarks, auto)) |
210 |
|
|
211 |
|
# The title is constructed from the titles of the input tables. |
212 |
|
self.assertEquals(table.Title(), |
213 |
|
"Join of %s and %s" % (landmarks.Title(), |
214 |
|
auto.Title())) |
215 |
|
|
216 |
|
|
217 |
|
def test_transient_joined_table_same_column_name(self): |
218 |
|
"""Test TransientJoinedTable join on columns with same name |
219 |
|
|
220 |
|
The transient DB maps the column names used by the tables to |
221 |
|
another set of names used only inside the SQLite database. There |
222 |
|
was a bug in the way this mapping was used when joining on |
223 |
|
fields with the same names in both tables so that the joined |
224 |
|
table ended up joining on the same column in the same table. |
225 |
|
""" |
226 |
|
mem_stretches = MemoryTable([("stretch_id", FIELDTYPE_INT)], |
227 |
|
[(i,) for i in range(4)]) |
228 |
|
stretches = AutoTransientTable(self.transientdb, mem_stretches) |
229 |
|
|
230 |
|
mem_discharges = MemoryTable([("disch_id", FIELDTYPE_INT), |
231 |
|
("stretch_id", FIELDTYPE_INT)], |
232 |
|
[(1, 0), (2, 3)]) |
233 |
|
discharges = AutoTransientTable(self.transientdb, mem_discharges) |
234 |
|
|
235 |
|
table = TransientJoinedTable(self.transientdb, stretches, "stretch_id", |
236 |
|
discharges, "stretch_id", |
237 |
|
outer_join = True) |
238 |
|
|
239 |
|
self.assertEquals(table.NumRows(), 4) |
240 |
|
self.assertEquals(table.NumColumns(), 2) |
241 |
|
|
242 |
|
# HasColumn |
243 |
|
self.failUnless(table.HasColumn("stretch_id")) |
244 |
|
self.failUnless(table.HasColumn("disch_id")) |
245 |
|
|
246 |
|
|
247 |
|
def test_transient_joined_table_with_equal_column_names(self): |
248 |
|
"""Test TransientJoinedTable join on tables with equal column names |
249 |
|
|
250 |
|
The join of two tables contains all fields from both tables instead |
251 |
|
th field the join was performed on. This special field is included |
252 |
|
once. If a name collision occurs for the field names, underscores are |
253 |
|
appended as long as any collision is resolved. |
254 |
|
""" |
255 |
|
mem_stretches = MemoryTable([("stretch_id", FIELDTYPE_INT), |
256 |
|
("name", FIELDTYPE_INT)], |
257 |
|
[(0, 10), (1, 11), (2, 12), (3, 13) ]) |
258 |
|
stretches = AutoTransientTable(self.transientdb, mem_stretches) |
259 |
|
|
260 |
|
mem_discharges = MemoryTable([("disch_id", FIELDTYPE_INT), |
261 |
|
("stretch_id", FIELDTYPE_INT), |
262 |
|
("name", FIELDTYPE_INT)], |
263 |
|
[(1, 0, 1), (2, 3, 2)]) |
264 |
|
discharges = AutoTransientTable(self.transientdb, mem_discharges) |
265 |
|
|
266 |
|
table = TransientJoinedTable(self.transientdb, stretches, "stretch_id", |
267 |
|
discharges, "stretch_id", |
268 |
|
outer_join = True) |
269 |
|
|
270 |
|
self.assertEquals(table.NumRows(), 4) |
271 |
|
self.assertEquals(table.NumColumns(), 4) |
272 |
|
|
273 |
|
# HasColumn |
274 |
|
self.failUnless(table.HasColumn("stretch_id")) |
275 |
|
self.failUnless(table.HasColumn("disch_id")) |
276 |
|
self.failUnless(table.HasColumn("name")) |
277 |
|
self.failUnless(table.HasColumn("name_")) |
278 |
|
|
279 |
|
|
280 |
def test_transient_table_read_twice(self): |
def test_transient_table_read_twice(self): |
281 |
"""Test TransientTable.ReadRowAsDict() reading the same record twice""" |
"""Test TransientTable.ReadRowAsDict() reading the same record twice""" |