def test_biomes(): terrain = Terrain(CONFIG) height_map = np.asarray([[ 0.5, 0.60000000001, 0.7, 0.799999999999999, 0.631, 0.95, 0.96, 0.81, 1. ]], dtype=float) moisture_map = np.asarray( [[1., 0.111, 0., 0.6, 0.71, 0.4, 0.69, 0.71, 0.1]], dtype=float) terrain.moisture_map = np.transpose(moisture_map) terrain.height_map = np.transpose(height_map) biome_map = terrain.get_biome_map() assert biome_map[0][0] == 'WATER' assert biome_map[1][0] == 'BEACH' assert biome_map[2][0] == 'DESERT' assert biome_map[3][0] == 'GRASSLAND' assert biome_map[4][0] == 'FOREST' assert biome_map[5][0] == 'GRASSLAND' assert biome_map[6][0] == 'TAIGA' assert biome_map[7][0] == 'SNOW' assert biome_map[8][0] == 'SNOW'