class TestMountain(unittest.TestCase):
def __init__(self, *args, **kwargs):
super(TestMountain, self).__init__(*args, **kwargs)
self.m1 = Mountain(10, 0)
def test_gen_img(self):
m2 = Mountain(100, 0, (50, 50))
hm = HeightMap(100, 100, m2.z)
hm.export("mountain_test.png")
def test_z(self):
z = self.m1.z((0, 0))
out_of_radius_1 = self.m1.center_pos + numpy.array((self.m1.radius + 1, 0))
out_of_radius_2 = self.m1.center_pos + numpy.array((0, self.m1.radius + 1))
out_of_radius_3 = self.m1.center_pos + numpy.array((1, self.m1.radius))
self.assertEqual(self.m1.z(out_of_radius_1), 0)
self.assertEqual(self.m1.z(out_of_radius_2), 0)
self.assertEqual(self.m1.z(out_of_radius_3), 0)
def test_influence_weight(self):
self.assertEqual(self.m1.influence((0, 0)), 1)
def __init__(self, *args, **kwargs):
super(TestMountain, self).__init__(*args, **kwargs)
self.m1 = Mountain(10, 0)
