class TestMountainImg(unittest.TestCase): def __init__(self, *args, **kwargs): super(TestMountainImg, self).__init__(*args, **kwargs) self.m1 = MountainImg(geom.Polygon([(0, 0), (10, 0), (10, 10), (0, 10)]), path="models/mountains/") def test_gen_img(self): hm = HeightMap(100, 100, self.m1.z) hm.export("mountain_img.png") def test_z(self): z = self.m1.z((0, 0)) out_of_radius_1 = self.m1.center_pos + numpy.array(self.m1.bb[2:4] + numpy.array((3, 3))) self.assertEqual(self.m1.z(out_of_radius_1), 0) def test_influence_weight(self): self.assertEqual(self.m1.influence((0, 0)), 1)
def __init__(self, *args, **kwargs): super(TestMountainImg, self).__init__(*args, **kwargs) self.m1 = MountainImg(geom.Polygon([(0, 0), (10, 0), (10, 10), (0, 10)]), path="models/mountains/")