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)