def testLinearSolverSolveForY(self, x1, y1, x2, y2, x, y): solver = clustering_centroids.TFLinearEquationSolver(float(x1), float(y1), float(x2), float(y2)) for_y = solver.solve_for_y(x) self.assertAlmostEqual(K.batch_get_value([for_y])[0], y)
def testLinearSolverConstruction(self, x1, y1, x2, y2, a, b): """Verifies that a TFLinearEquationSolver is constructed correctly.""" solver = clustering_centroids.TFLinearEquationSolver( float(x1), float(y1), float(x2), float(y2)) solver_a = solver.a self.assertAlmostEqual(K.batch_get_value([solver_a])[0], a) self.assertAlmostEqual(K.batch_get_value([solver.b])[0], b)
def testLinearSolverSolveForX(self, x1, y1, x2, y2, x, y): """Verifies that TFLinearEquationSolver solves the given equations correctly for X.""" solver = clustering_centroids.TFLinearEquationSolver(float(x1), float(y1), float(x2), float(y2)) for_x = solver.solve_for_x(y) self.assertAlmostEqual(K.batch_get_value([for_x])[0], x)