def test_is_finished(self): executor = SimplexExecutor() finished = executor.is_finished( np.array([[1.0, 0], [1.0, 0], [1.0, 0]]), # vertices np.array([1, 1, 1]), # vertex ranks 2, # reflection rank 2, # expansion rank 2, # contraction rank ) self.assertTrue(finished)
def test_update_points(self): executor = SimplexExecutor() updated_vertices = executor.update_points( np.array([[1.0, 0], [2.0, 0], [2.9, 0]]), # vertices np.array([1, 2, 3]), # vertex ranks 0, # reflection rank 4, # expansion rank 5, # contraction rank ) self.assertTrue(np.all( updated_vertices - np.array([[1.0, 0], [2.0, 0], [float(31)/30, 0]]) < 0.0001 ))
def test_get_next_points(self): executor = SimplexExecutor() reflection, expansion, contraction = executor.get_next_points( np.array([[1.0, 0], [2.0, 0], [2.9, 0]]), # vertices np.array([1, 2, 3]), ) self.assertTrue(np.all( reflection - np.array([float(31)/30, 0]) < 0.0001 )) self.assertTrue(np.all( expansion - np.array([float(2)/15, 0]) < 0.0001 )) self.assertTrue(np.all( contraction - np.array([float(37)/15, 0]) < 0.0001 ))