Fix #227, python list versus numpy array in test_loom.

probcomp · Nov 16, 2017 · 3c751d9 · 3c751d9
1 parent 79dc561
commit 3c751d9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/disabled_test_loomcat.py b/tests/disabled_test_loomcat.py
@@ -181,7 +181,9 @@ def test_dependence_probability():
     logscore1 = engine.logpdf_score()
     assert numpy.mean(logscore1) > numpy.mean(logscore0)
 
-    dependence_probability = engine.dependence_probability_pairwise()
+    dependence_probability = numpy.mean(
+        engine.dependence_probability_pairwise(),
+        axis=0)
 
     assert dependence_probability[0,1] > 0.8
     assert dependence_probability[1,2] > 0.8