def test_to_from_json(self):
"""Test FlatTable from_dict method"""
json_string = self.flat_table.to_json()
flat_table = FlatTable()
flat_table.from_json(json_string)
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)
def test_from_json(self):
"""Test FlatTable from_json method"""
json_string = (
'{"data": [["residential", "low", 50], '
'["residential", "medium", 30], ["secondary", "low", 40], '
'["primary", "high", 10], ["primary", "medium", 20]], '
'"groups": ["road_type", "hazard"]}')
flat_table = FlatTable()
flat_table.from_json(json_string)
expected_groups = ["road_type", "hazard"]
for i in range(len(flat_table.groups)):
self.assertEquals(expected_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)
