def test_single_aggregation_trees(self):
aggregation_tree = {"org_chart": ["CEO", "Architect", "Developer"]}
entity = Entity(geocode="1234", geoname="Accra", unique_name="Kajelo CHPS", aggregation_tree=aggregation_tree)
entity.save()
loaded_entity = query.get(uuid=entity.uuid)
self.assertEqual(loaded_entity.aggregation_tree["org_chart"], ["CEO", "Architect", "Developer"])
def test_load_entity(self):
entity = Entity(geocode="123466", geoname="Ghana", unique_name="Navio CHPS")
entity.save()
loaded_entity = query.get(uuid=entity.uuid)
self.assertEqual(loaded_entity.uuid, entity.uuid)