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)
def test_pickle(self):
a = Environment([Feature()])
# moche mais en attendant que ça marche...
a.init_heightmap(10, 10)
b = HeightMap(10, 10, z_func=lambda z: 100 * z[0] + z[1])
a.heightmap = b
f = open('/tmp/waves.p', 'wb')
pickle.dump(a, f)
f.close()
g = open('/tmp/waves.p', 'rb')
b = pickle.load(g)
g.close()
self.assertEqual(a.heightmap, b.heightmap)