def test_create_incorrect_length(self):
self.y = np.array([1])
with self.assertRaises(
ValueError,
msg="Dimension 0 of x and y do not match. x: {0}, y: {1}".
format(self.x.shape, self.y.shape)):
Data(x=self.x, y=self.y)
def test_index_2(self):
self.y = np.array([8, 4, 5])
data = Data(x=self.x, y=self.y)
x, y = data[[0, 2]]
self.assertTrue((x == self.x[[0, 2], :]).all())
self.assertTrue((y == self.y[[0, 2]]).all())
def test_index_1(self):
self.y = np.array([8, 4, 5])
data = Data(x=self.x, y=self.y)
x, y = data[1]
self.assertTrue((x == self.x[1, :]).all())
self.assertTrue((y == self.y[1]).all())
def test_create_not_iterable(self):
self.y = 'cat'
with self.assertRaises(AttributeError, msg="Data must be iterable."):
Data(x=self.x, y=self.y)
def test_create_no_shape(self):
self.y = 'cat'
with self.assertRaises(AttributeError,
msg="Data must have a 'shape' attribute."):
Data(x=self.x, y=self.y)
def test_create_correct(self):
self.y = np.array([8, 4, 5])
Data(x=self.x, y=self.y)
def test_create_empty(self):
Data()