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)
