def test_get_plot_coeffs(self): # Test if x * p1 + y * p2 yields p3. c = TEMCalculator() coeffs = c.get_plot_coeffs((1, 1, 0), (1, -1, 0), (2, 0, 0)) self.assertArrayAlmostEqual(np.array([1.0, 1.0]), coeffs)
def test_get_plot_coeffs(self): # Test if x * p1 + y * p2 yields p3. c = TEMCalculator() coeffs = c.get_plot_coeffs((1, 1, 0), (1, -1, 0), (2, 0, 0), -2, False) self.assertEqual([1, 1], coeffs)