diff --git a/test/baker2d.py b/test/baker2d.py index d3a2627..5123262 100644 --- a/test/baker2d.py +++ b/test/baker2d.py @@ -98,9 +98,7 @@ class Test(unittest.TestCase): answer = baker.run_baker(self.X, R, S) - a = answer['abc'][0] - b = answer['abc'][1] - c = answer['abc'][2] + a, b, c = sorted(answer['abc']) aa,bb,cc = sorted((2/3.0, 2/3.0, 1/3.0)) self.assertAlmostEqual(a,aa)