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)