def test_from_dict(self):
"""Test FlatTable from_dict method"""
groups = ["road_type", "hazard"]
data = [
["residential", "low", 50],
["residential", "medium", 30],
["secondary", "low", 40],
["primary", "high", 10],
["primary", "medium", 20]
]
flat_table = FlatTable()
flat_table.from_dict(groups, data)
for i in range(len(flat_table.groups)):
self.assertEquals(groups[i], flat_table.groups[i])
self.assertEquals(flat_table.data[('residential', 'low')], 50)
self.assertEquals(flat_table.data[('residential', 'medium')], 30)
self.assertEquals(flat_table.data[('secondary', 'low')], 40)
self.assertEquals(flat_table.data[('primary', 'high')], 10)
self.assertEquals(flat_table.data[('primary', 'medium')], 20)
groups = [u'landcover', u'hazard', u'zone']
data = [
[u'Forest', u'high', None, 5172.100048073517],
[u'Population', u'high', None, 20689.8283632199],
[u'Forest', u'low', None, 5171.381989317935],
[u'Population', u'medium', None, 10347.048486941067],
[u'Meadow', u'high', None, 5172.81413353821],
[u'Population', u'low', None, 10342.763978632318],
[u'Meadow', u'medium', None, 5173.5242434723095]
]
flat_table = FlatTable()
flat_table.from_dict(groups, data)
for i in range(len(flat_table.groups)):
self.assertEquals(groups[i], flat_table.groups[i])
self.assertEquals(flat_table.data[
('Forest', 'high', None)], 5172.100048073517)
