1 |
# Copyright (c) 2002, 2003 by Intevation GmbH |
# Copyright (c) 2002, 2003, 2006 by Intevation GmbH |
2 |
# Authors: |
# Authors: |
3 |
# Bernhard Herzog <[email protected]> |
# Bernhard Herzog <[email protected]> |
4 |
# |
# |
19 |
import support |
import support |
20 |
support.initthuban() |
support.initthuban() |
21 |
|
|
22 |
|
import dbflib |
23 |
from Thuban.Model.table import DBFTable, MemoryTable, FIELDTYPE_STRING, \ |
from Thuban.Model.table import DBFTable, MemoryTable, FIELDTYPE_STRING, \ |
24 |
FIELDTYPE_INT, FIELDTYPE_DOUBLE |
FIELDTYPE_INT, FIELDTYPE_DOUBLE, table_to_dbf |
25 |
from Thuban.Model.transientdb import TransientDatabase, TransientTable, \ |
from Thuban.Model.transientdb import TransientDatabase, TransientTable, \ |
26 |
TransientJoinedTable, AutoTransientTable |
TransientJoinedTable, AutoTransientTable |
27 |
|
|
43 |
self.transientdb.close() |
self.transientdb.close() |
44 |
|
|
45 |
def run_iceland_political_tests(self, table): |
def run_iceland_political_tests(self, table): |
46 |
"""Run some tests on tablte |
"""Run some tests on table |
47 |
|
|
48 |
Assume that table holds the data of the file |
Assume that table holds the data of the file |
49 |
../Data/iceland/political.dbf sample file. |
../Data/iceland/political.dbf sample file. |
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.ReadRowAsDict(144, row_is_ordinal = 1), |
78 |
|
{'POPYCOUN': 'IC', 'POPYADMIN': '', 'PONET_': 146, |
79 |
|
'AREA': 19.462, |
80 |
|
'POPYTYPE': 1, 'PERIMETER': 88.518000000000001, |
81 |
|
'POPYREG': '1', |
82 |
|
'PONET_ID': 145}) |
83 |
self.assertEquals(table.ReadValue(144, "AREA"), 19.462) |
self.assertEquals(table.ReadValue(144, "AREA"), 19.462) |
84 |
self.assertEquals(table.ReadValue(144, 3), 145) |
self.assertEquals(table.ReadValue(144, 3), 145) |
85 |
|
self.assertEquals(table.ReadValue(144, "AREA", row_is_ordinal = 1), |
86 |
|
19.462) |
87 |
|
self.assertEquals(table.ReadValue(144, 3, row_is_ordinal = 1), 145) |
88 |
|
|
89 |
|
self.assertEquals(table.RowIdToOrdinal(23), 23) |
90 |
|
self.assertEquals(table.RowOrdinalToId(23), 23) |
91 |
|
|
92 |
# ValueRange may induce a copy to the transient database. |
# ValueRange may induce a copy to the transient database. |
93 |
# Therefore we put it last so that we can execute this method |
# Therefore we put it last so that we can execute this method |
136 |
# least one call to a method that copies to the transient db at |
# least one call to a method that copies to the transient db at |
137 |
# its end. |
# its end. |
138 |
self.run_iceland_political_tests(table) |
self.run_iceland_political_tests(table) |
139 |
|
# At this point the data has probably not been copied to the |
140 |
|
# transient DB yet, so we force it by calling the |
141 |
|
# transient_table method. |
142 |
|
table.transient_table() |
143 |
|
|
144 |
|
# Run the tests again. |
145 |
self.run_iceland_political_tests(table) |
self.run_iceland_political_tests(table) |
146 |
|
|
147 |
def test_auto_transient_table_query(self): |
def test_auto_transient_table_query(self): |
220 |
'code': 1, 'type': 'RUINS'}) |
'code': 1, 'type': 'RUINS'}) |
221 |
self.assertEquals(table.ReadValue(22, "type"), 'RUINS') |
self.assertEquals(table.ReadValue(22, "type"), 'RUINS') |
222 |
self.assertEquals(table.ReadValue(22, 7), 1) |
self.assertEquals(table.ReadValue(22, 7), 1) |
223 |
|
self.assertEquals(table.ReadValue(22, "type", row_is_ordinal = 1), |
224 |
|
"RUINS") |
225 |
|
self.assertEquals(table.ReadValue(22, 7, row_is_ordinal = 1), 1) |
226 |
|
self.assertEquals(table.RowIdToOrdinal(23), 23) |
227 |
|
self.assertEquals(table.RowOrdinalToId(23), 23) |
228 |
|
|
229 |
# The transient_table method should return the table itself |
# The transient_table method should return the table itself |
230 |
self.assert_(table is table.transient_table()) |
self.assert_(table is table.transient_table()) |
385 |
table.SimpleQuery, |
table.SimpleQuery, |
386 |
table.Column(1), "<<", table.Column(2)) |
table.Column(1), "<<", table.Column(2)) |
387 |
|
|
388 |
|
def test_transienttable_to_dbf(self): |
389 |
|
memtable = MemoryTable([("type", FIELDTYPE_STRING), |
390 |
|
("value", FIELDTYPE_DOUBLE), |
391 |
|
("code", FIELDTYPE_INT)], |
392 |
|
[("UNKNOWN", 0.0, 0), |
393 |
|
("Foo", 0.5, -1), |
394 |
|
("Foo", 1.0/256, 100), |
395 |
|
("bar", 1e10, 17)]) |
396 |
|
table = TransientTable(self.transientdb, memtable) |
397 |
|
filename = self.temp_file_name("test_transienttable_to_dbf.dbf") |
398 |
|
table_to_dbf(table, filename) |
399 |
|
|
400 |
|
dbf = dbflib.DBFFile(filename) |
401 |
|
self.assertEquals(dbf.read_record(2), |
402 |
|
{'code': 100, 'type': 'Foo', 'value': 0.00390625}) |
403 |
|
self.assertEquals(dbf.field_count(), 3) |
404 |
|
self.assertEquals(dbf.record_count(), 4) |
405 |
|
self.assertEquals(dbf.field_info(0), |
406 |
|
(dbflib.FTString, "type", 7, 0)) |
407 |
|
self.assertEquals(dbf.field_info(1), |
408 |
|
(dbflib.FTDouble, "value", 24, 12)) |
409 |
|
self.assertEquals(dbf.field_info(2), |
410 |
|
(dbflib.FTInteger, "code", 3, 0)) |
411 |
|
|
412 |
|
|
413 |
|
|
414 |
if __name__ == "__main__": |
if __name__ == "__main__": |
415 |
support.run_tests() |
support.run_tests() |