def test_mountains(self):
mount = ImageFeature("tests/img/mount_200.png")
gen_model = AbstractModel(os.path.abspath(os.path.dirname(sys.argv[0])) + "/tests/models/pine_tree/Pine_4m.obj", 0.01, (0, 0))
v1 = Vegetation(geom.box(20, 20, 180, 180), gen_model, 10)
v1.models.append(Model((5, 5), gen_model))
v1.models.append(Model((195, 195), gen_model))
env = Environment([mount, v1])
env.init_heightmap(200, 200)
f = open('mountains.pickle', 'wb')
pickle.dump(env, f)
f.close()
env.export_heightmap("bla.png", 200, 200)
class TestClassEnvironment(unittest.TestCase):
def setUp(self):
self.background = FeatureTest(100, influence="notall", val_influence=1)
self.background.shape = geom.box(5, 5, 45, 45)
self.up = FeatureTest(150, influence="notall", val_influence=1)
self.up.shape = geom.box(10, 10, 30, 30)
self.rep = FeatureTestReplace(0, influence="notall", val_influence=1)
self.rep.shape = geom.box(10, 10, 30, 30)
self.forest = Vegetation(geom.box(5, 5, 45, 45), model=AbstractModel(), tree_number=50)
self.env = Environment([self.rep, self.up, self.background, self.forest])
def test_models(self):
self.assertLess(len(self.env.models), 50)
self.assertGreater(len(self.env.models), 25)
self.env.export_heightmap("temp_env.png", 50, 50)
def test_mountain(self):
m2 = Mountain(10**3, 0, (50, 50))
t = Environment([m2])
t.export_heightmap("mountain_as_env.png", res_x=100, res_y=100)