[Predator] Making the Y sets conditional for loglinear models

appengine/findit/crash/loglinear/test/model_test.py

Index: appengine/findit/crash/loglinear/test/model_test.py
diff --git a/appengine/findit/crash/loglinear/test/model_test.py b/appengine/findit/crash/loglinear/test/model_test.py
index e80031c26ed449376fbdac8237b4282eb87e985b..17d8e6e4b85a3b0a2cf9bb9e235042f119b6f750 100644
--- a/appengine/findit/crash/loglinear/test/model_test.py
+++ b/appengine/findit/crash/loglinear/test/model_test.py
@@ -15,7 +15,7 @@ class LoglinearTest(LoglinearTestCase):
   def testToFeatureFunction(self):
     """Test that ``ToFeatureFunction`` obeys the equality its docstring says."""
     for x in self._X:
-      for y in self._Y:
+      for y in self._Y(x):
         for i in xrange(self._qty_features):
           self.assertEqual(self._feature_list[i](x)(y),
                            self._feature_function(x)(y)[i])
@@ -43,7 +43,7 @@ class LoglinearTest(LoglinearTestCase):
     for x in self._X:
       self.assertEqual(math.exp(model.LogZ(x)), model.Z(x))
       model.Expectation(x, lambda y: np.array([1.0]))
-      for y in self._Y:
+      for y in self._Y(x):
         model.Features(x)(y)
         model.Score(x)(y)
         self.assertEqual(