class SimplexUtilityTest(unittest.TestCase):
def setUp(self):
self.simplex = Simplex()
def test_get_centroid(self):
c = self.simplex.centroid(np.array([[1, 1], [3, 3], [2, 2]]))
self.assertEqual(len(c), 2)
self.assertTrue(np.all(c == np.array([2, 2])))
def test_get_ranks(self):
input_mock = mock.Mock(return_value="2 1")
with mock.patch("__builtin__.raw_input", input_mock):
ranks = query_ranks(np.array([[1, 1], [2, 2]]))
self.assertTrue(np.all(ranks == np.array([2, 1])))
def test_get_ranks_skip_extraneous_input(self):
input_mock = mock.Mock(return_value="2 1 3 4 5")
with mock.patch("__builtin__.raw_input", input_mock):
ranks = query_ranks(np.array([[1, 1], [2, 2]]))
self.assertTrue(np.all(ranks == np.array([2, 1])))
