def test_points_selection_05(self):
nc = [5]
x = np.array([[8.79], [0.35], [4.56]])
data_feed = UniformSampling(self.test_data,
list_of_samples_per_variable=nc)
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_06(self):
output_2 = UniformSampling(self.test_data, [3])
expected_output = 3
expected_output_2 = np.array([[0], [1], [2], [3], [4], [5], [6], [7],
[8], [9]])
expected_output_3 = [3]
expected_output_4 = self.test_data.columns.values.tolist()
self.assertEqual(output_2.number_of_samples, expected_output)
self.assertEqual(output_2.x_data.tolist(), expected_output_2.tolist())
self.assertEqual(output_2.dim_vector, expected_output_3)
self.assertEqual(output_2.data_headers, expected_output_4)
def test_initialization_12(self):
test_data_as_list = self.test_data.values.tolist()
with self.assertRaises(ValueError):
UniformSampling(test_data_as_list, [7])
def test_initialization_11(self):
test_data_as_array = self.test_data.values
output = UniformSampling(test_data_as_array, [7])
np.testing.assert_array_equal(output.data, test_data_as_array)
self.assertEqual(output.number_of_samples, 7)
def test_initialization_10(self):
output = UniformSampling(self.test_data, [3], edges=False)
self.assertEqual(output.edge, False)
def test_initialization_09(self):
output = UniformSampling(self.test_data, [3], edges=True)
self.assertEqual(output.edge, True)
def test_initialization_07(self):
with self.assertRaises(Exception):
UniformSampling(self.test_data, [3], edges='x')
def test_initialization_05(self):
with self.assertRaises(Exception):
UniformSampling(self.test_data, [11])
def test_initialization_04(self):
with self.assertRaises(TypeError):
UniformSampling(self.test_data, [2.3])
def test_initialization_03(self):
with self.assertRaises(ValueError):
UniformSampling(self.test_data, [1])
def test_initialization_01(self):
with self.assertRaises(TypeError):
UniformSampling(self.test_data, np.array([2]))