removes transpose and adds correct matrix

quantling · Apr 19, 2017 · acd6b93 · acd6b93
1 parent 98f0301
commit acd6b93
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_activation.py b/tests/test_activation.py
@@ -46,7 +46,7 @@ def test_activation_matrix():
               (['c1', 'c3'], []),
               (['c2'], []),
               (['c1', 'c1'], [])]
-    reference_activations = np.array([[1, 1], [0, 1], [1, 0], [0, 1]]).T
+    reference_activations = np.array([[1, 0, 1, 0], [1, 1, 0, 1]])
 
     with pytest.raises(ValueError):
         activations = activation(events, weights, number_of_threads=1)
@@ -70,7 +70,7 @@ def test_ignore_missing_cues():
               (['c1', 'c3'], []),
               (['c2', 'c4'], []),
               (['c1', 'c1'], [])]
-    reference_activations = np.array([[1, 1], [0, 1], [1, 0], [0, 1]]).T
+    reference_activations = np.array([[1, 0, 1, 0], [1, 1, 0, 1]])
 
     with pytest.raises(KeyError):
         activations = activation(events, weights, number_of_threads=1,