def plot(): equation_string = flask.request.args.get('equation') diff_equation = FieldPlotter() diff_equation.set_equation_from_string(equation_string) diff_equation.make_plot() # If plotting was successful, write plot out if diff_equation.figure: # Write output to memory and add to response object output = StringIO.StringIO() response = flask.make_response(base64.b64encode(diff_equation.write_data(output))) response.mimetype = 'image/png' return response else: return flask.make_response('')
def test_make_plot(self): equation = sympy.sympify('x+y') plotter = FieldPlotter(equation) plotter.make_plot() self.assertIsNotNone(plotter.figure)
def test_missing_equation(self): plotter = FieldPlotter() with self.assertRaises(FieldPlotter.MissingEquationError): plotter.make_plot()