def river_test(): terrain = Terrain() terrain.generate_terrain(num_rivers=0) z, x = terrain.layers['material'].shape sp1, ep1 = (0, x//2), (z//2 -5, x//2) sp2, ep2 = (z//2 + 5, x//2), (z-1, x//2) terrain.add_material_line(terrain.layers['material'], terrain.materials['water'], sp1, ep1, 1) terrain.add_material_line(terrain.layers['material'], terrain.materials['water'], sp2, ep2, 1) start_road = (random.randint(0, z-1), 0) end_road = (random.randint(0, z-1), x-1) terrain.add_road([end_road]) connect_point(start_road, terrain) print('start road', start_road) print('end road', end_road) plot(terrain, [])