29 |
from Thuban.Model.color import Color |
from Thuban.Model.color import Color |
30 |
from Thuban.Model.table import MemoryTable, FIELDTYPE_STRING, \ |
from Thuban.Model.table import MemoryTable, FIELDTYPE_STRING, \ |
31 |
FIELDTYPE_INT, FIELDTYPE_DOUBLE |
FIELDTYPE_INT, FIELDTYPE_DOUBLE |
32 |
|
from Thuban.Model.data import DerivedShapeStore |
33 |
|
|
34 |
|
|
35 |
class TestSessionSimple(unittest.TestCase): |
class TestSessionSimple(unittest.TestCase): |
36 |
|
|
71 |
table = session.AddTable(memtable) |
table = session.AddTable(memtable) |
72 |
self.assertEquals(session.Tables(), [table]) |
self.assertEquals(session.Tables(), [table]) |
73 |
|
|
74 |
|
def test_remove_table(self): |
75 |
|
"""Test Session.RemoveTable()""" |
76 |
|
session = self.session = Session("Test Session") |
77 |
|
memtable = MemoryTable([("type", FIELDTYPE_STRING), |
78 |
|
("value", FIELDTYPE_DOUBLE), |
79 |
|
("code", FIELDTYPE_INT)], |
80 |
|
[("OTHER/UNKNOWN", -1.5, 11), |
81 |
|
("RUINS", 0.0, 1), |
82 |
|
("FARM", 3.141, 2), |
83 |
|
("BUILDING", 2.5, 3), |
84 |
|
("HUT", 1e6, 4), |
85 |
|
("LIGHTHOUSE", -0.01, 5)]) |
86 |
|
table = session.AddTable(memtable) |
87 |
|
self.assertEquals(session.Tables(), [table]) |
88 |
|
session.RemoveTable(table) |
89 |
|
self.assertEquals(session.Tables(), []) |
90 |
|
self.assertRaises(ValueError, session.RemoveTable, table) |
91 |
|
|
92 |
def test_open_shapefile(self): |
def test_open_shapefile(self): |
93 |
"""Test Session.OpenShapefile()""" |
"""Test Session.OpenShapefile()""" |
94 |
session = self.session = Session("Test Session") |
session = self.session = Session("Test Session") |
105 |
|
|
106 |
self.assertEquals(session.Tables(), [store.Table()]) |
self.assertEquals(session.Tables(), [store.Table()]) |
107 |
|
|
108 |
|
def test_add_shapestore(self): |
109 |
|
"""Test Session.AddShapeStore()""" |
110 |
|
session = self.session = Session("Test Session") |
111 |
|
filename = os.path.join("..", "Data", "iceland", |
112 |
|
"roads-line.shp") |
113 |
|
try: |
114 |
|
store = session.OpenShapefile(filename) |
115 |
|
derived = DerivedShapeStore(store, store.Table()) |
116 |
|
session.AddShapeStore(derived) |
117 |
|
self.assertEquals(session.ShapeStores(), [store, derived]) |
118 |
|
finally: |
119 |
|
store = derived = None |
120 |
|
|
121 |
|
|
122 |
class TestSessionBase(unittest.TestCase, support.SubscriberMixin): |
class TestSessionBase(unittest.TestCase, support.SubscriberMixin): |
123 |
|
|