Exemplo n.º 1
0
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'