Fixing tests (again)

Author: perimosocordiae

test/metric_learn_test.py

@@ -75,10 +75,10 @@ class TestSDML(MetricTestCase):
   def test_iris(self):
     # Note: this is a flaky test, which fails for certain seeds.
     # TODO: un-flake it!
-    np.random.seed(5555)
+    rs = np.random.RandomState(5555)
 
     sdml = SDML_Supervised(num_constraints=1500)
-    sdml.fit(self.iris_points, self.iris_labels)
+    sdml.fit(self.iris_points, self.iris_labels, random_state=rs)
     csep = class_separation(sdml.transform(), self.iris_labels)
     self.assertLess(csep, 0.25)
 

test/test_fit_transform.py

@@ -92,7 +92,7 @@ def test_lfda(self):
     res_2 = lfda.fit_transform(self.X, self.y)
 
     # signs may be flipped, that's okay
-    if res_1[0,0] != res_2[0,0]:
+    if np.sign(res_1[0,0]) != np.sign(res_2[0,0]):
         res_2 *= -1
     assert_array_almost_equal(res_1, res_2)