def setup(self): super(TestBuilder, self).setup() import builder from builder.builder import Builder entitynames = Builder._get_entity_names() terrain_names = [str(t) for t in Builder._get_terrain_names()] self.entity_class1 = new_entity_class(entitynames[0], terrain_names) self.entity_class2 = new_entity_class(entitynames[1], terrain_names)
def test_get_terrain_names(self): """ Récupérer les noms des entités""" b = Builder() terrains = [str(s) for s in b._get_terrain_names()] assert "Montagne" in terrains assert "Zone_Risque" in terrains