Make weights for test deterministic.

lmjohns3 · Jun 9, 2015 · ddac2cc · ddac2cc
1 parent 105dd97
commit ddac2cc
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/test/feedforward_test.py b/test/feedforward_test.py
@@ -82,9 +82,8 @@ def _build(self, *hiddens):
 
     def test_score_onelayer(self):
         net = self._build(13)
-        z = net.score(self.images,
-                      self.labels,
-                      np.random.randint(0, 2, size=self.labels.shape))
+        z = net.score(
+            self.images, self.labels, 0.5 * np.ones(self.labels.shape, 'f'))
         assert 0 < z < 1