def test_points_selection_02(self): number_of_samples = 5 x = np.array([[8.79], [0.35], [4.56]]) data_feed = HammersleySampling(self.test_data, number_of_samples) output = data_feed.points_selection(self.test_data.values, x) expected_output = np.array([[9., 100.], [0., 1.], [5., 36.]]) self.assertEqual(output.tolist(), expected_output.tolist())
def test_initialization_05(self): output_2 = HammersleySampling(self.test_data) expected_output = 5 expected_output_2 = np.array([[0], [1], [2], [3], [4], [5], [6], [7], [8], [9]]) self.assertEqual(output_2.number_of_samples, expected_output) self.assertEqual(output_2.x_data.tolist(), expected_output_2.tolist())
def test_data_sequencing_03(self): data_feed = HammersleySampling(self.test_data, number_of_samples=5) output = data_feed.data_sequencing(no_samples=5, prime_base=5) expected_output = np.array([0, 0.2, 0.4, 0.6, 0.8]) np.testing.assert_array_equal(expected_output, output)
def test_initialization_08(self): test_data_as_list = self.test_data.values.tolist() with self.assertRaises(ValueError): HammersleySampling(test_data_as_list, 7)
def test_initialization_07(self): test_data_as_array = self.test_data.values output_1 = HammersleySampling(test_data_as_array, 7) np.testing.assert_array_equal(output_1.data, test_data_as_array)
def test_initialization_06(self): with self.assertRaises(Exception): HammersleySampling(self.test_data, 1.7)