def test_points_selection_03(self):
number_of_samples = 5
x = np.array([[8.79], [0.35], [4.56]])
data_feed = HaltonSampling(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 = HaltonSampling(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_02(self):
data_feed = HaltonSampling(self.test_data, number_of_samples=5)
output = data_feed.data_sequencing(no_samples=1, prime_base=3)
expected_output = np.array([0])
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):
HaltonSampling(test_data_as_list, 7)
def test_initialization_07(self):
test_data_as_array = self.test_data.values
output_1 = HaltonSampling(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):
HaltonSampling(self.test_data, 1.7)