def test_random_state(self):
maps_A1, maps_B1 = maps.split(prop=0.5, random_state=0)
maps_A2, maps_B2 = maps.split(prop=0.5, random_state=0)
print(maps_A1.to_array())
print(maps_A2.to_array())
self.assertTrue(np.array_equal(maps_A1.to_array(), maps_A2.to_array()))
self.assertTrue(np.array_equal(maps_B1.to_array(), maps_B2.to_array()))
def test_half_prop_two_maps(self):
maps = Maps(self.array2)
maps_A, maps_B = maps.split(prop=0.5)
self.assertTrue(
np.array_equal(maps_A.to_array(), self.array2[:, :, :, 0])
or np.array_equal(maps_A.to_array(), self.array2[:, :, :, 1]))
self.assertTrue(
np.array_equal(maps_B.to_array(), self.array2[:, :, :, 0])
or np.array_equal(maps_B.to_array(), self.array2[:, :, :, 1]))
self.assertFalse(np.array_equal(maps_A.to_array(), maps_B.to_array()))
def test_min_prop_two_maps(self):
maps = Maps(self.array2)
maps_A, maps_B = maps.split(prop=0.)
self.assertTrue(np.array_equal(maps_B.to_array(), self.array2))
self.assertEqual(maps_A.n_maps, 0)
def test_max_prop_one_maps(self):
maps = Maps(self.array)
maps_A, maps_B = maps.split(prop=1.)
self.assertTrue(np.array_equal(maps_A.to_array(), self.array))
self.assertEqual(maps_B.n_maps, 0)