Fix tests for older Numpy versions.

nengo · Feb 7, 2018 · a2d0cf3 · a2d0cf3
1 parent 369018f
commit a2d0cf3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nengo_spa/testing.py b/nengo_spa/testing.py
@@ -9,7 +9,7 @@ def sp_close(
     actual = data[(skip < t) & (t <= skip + duration)]
     expected = target_sp
     if normalized:
-        actual /= np.linalg.norm(actual, axis=1, keepdims=True)
+        actual /= np.expand_dims(np.linalg.norm(actual, axis=1), 1)
         expected = expected.normalized()
     return np.all(np.sqrt(np.sum(
         np.square(actual - expected.v), axis=1)) < atol)