def setUp(self):
order = np.array([2, 0, 1])
permuted_matrix = np.array([[0, 0, 0],
[1, 0, 0],
[2, 3, 0]])
data = np.array([[0, 1, 2],
[0, 1, 2],
[0, 1, 2],
[0, 1, 2],
[0, 1, 2]])
labels = np.array(['a', 'b', 'c'])
self.r = Result(order=order, permuted_matrix=permuted_matrix, data=data, labels=labels)
class TestResultMethods(unittest.TestCase):
def setUp(self):
order = np.array([2, 0, 1])
permuted_matrix = np.array([[0, 0, 0],
[1, 0, 0],
[2, 3, 0]])
data = np.array([[0, 1, 2],
[0, 1, 2],
[0, 1, 2],
[0, 1, 2],
[0, 1, 2]])
labels = np.array(['a', 'b', 'c'])
self.r = Result(order=order, permuted_matrix=permuted_matrix, data=data, labels=labels)
def test_order(self):
self.assertIsNone(
np.testing.assert_array_equal(
self.r.order,
np.array([2, 0, 1])))
def test_matrix(self):
self.assertIsNone(
np.testing.assert_array_equal(
self.r.matrix,
np.array([[0, 0, 1],
[3, 0, 2],
[0, 0, 0]])))
def test_permuted_matrix(self):
self.assertIsNone(
np.testing.assert_array_equal(
self.r.permuted_matrix,
np.array([[0, 0, 0],
[1, 0, 0],
[2, 3, 0]])))
def test_data(self):
self.assertIsNone(
np.testing.assert_array_equal(
self.r.data,
np.array([[0, 1, 2],
[0, 1, 2],
[0, 1, 2],
[0, 1, 2],
[0, 1, 2]])))
def test_permuted_data(self):
self.assertIsNone(
np.testing.assert_array_equal(
self.r.permuted_data,
np.array([[2, 0, 1],
[2, 0, 1],
[2, 0, 1],
[2, 0, 1],
[2, 0, 1]])))
def test_labels(self):
self.assertIsNone(
np.testing.assert_array_equal(
self.r.labels,
np.array(['a', 'b', 'c'])
))
def test_permuted_labels(self):
self.assertIsNone(
np.testing.assert_array_equal(
self.r.permuted_labels,
np.array(['c', 'a', 'b'])))
def test_plot(self):
self.r.plot()
