class TestHeightMap(unittest.TestCase): def __init__(self, *args, **kwargs): super(TestHeightMap, self).__init__(*args, **kwargs) self.m1 = HeightMap(100, 100) self.m2 = HeightMap(100, 100, lambda z: z[0] + 1 + z[1]) def test_hmap(self): self.assertEqual(self.m1.hmap[0][0], 0) self.assertEqual(self.m2.hmap[0][0], 1) self.assertEqual(self.m2.hmap[1][2], 4) def test_export(self): self.m1.export("m1.png") self.m2.export("m2.png") self.m2.export("m3.png", 2) compare_imgs("tests/img/m1.png", "m1.png", self) compare_imgs("tests/img/m2.png", "m2.png", self) compare_imgs("tests/img/m3.png", "m3.png", self) def test_change_res(self): res = 2 self.m = HeightMap(10, 10, lambda z: z[0] + 10 * z[1]) self.assertEqual(self.m.hmap[9][9], 99) self.m.change_res(2) self.assertEqual(self.m.hmap[19][19], 99) self.assertEqual(self.m.hmap[19][18], 99) self.assertEqual(self.m.hmap[18][19], 99) self.assertEqual(self.m.hmap[18][18], 99)