def test_does_good_matrix_return_correct_size_array(self):
cmat = np.eye(2)
mu = np.zeros([2, 1])
ndp = 50 # number of points to generate ellipse shape
x, y = utilities.generate_ellipse(mu, cmat, ndp)
self.assertEqual(x.size, ndp)
self.assertEqual(y.size, ndp)
def test_does_non_positive_definite_matrix_return_error(self):
cmat = np.zeros([2, 2])
mu = np.zeros([2, 1])
with self.assertRaises(SystemExit):
utilities.generate_ellipse(mu, cmat)
def test_does_good_matrix_return_equal_sized_xy_arrays(self):
cmat = np.eye(2)
mu = np.zeros([2, 1])
x, y = utilities.generate_ellipse(mu, cmat)
self.assertEqual(x.shape, y.shape)
def test_does_non_symmetric_matrix_return_error(self):
cmat = np.array([[3, 2], [1, 3]])
mu = np.zeros([2, 1])
with self.assertRaises(SystemExit):
utilities.generate_ellipse(mu, cmat)
