Updating calls to NumPy stack to avoid generators.

neurophysik · Mar 6, 2020 · d702d96 · d702d96
1 parent 7cfab9a
commit d702d96
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/test_numerical.py b/tests/test_numerical.py
@@ -10,7 +10,7 @@ class RandomDirectionTest(unittest.TestCase):
 	def test_random_direction(self):
 		d = 10
 		n = 100000
-		n_vectors = np.vstack(random_direction(d) for i in range(n))
+		n_vectors = np.vstack([random_direction(d) for i in range(n)])
 		average = np.average(n_vectors, axis=0)
 		assert_allclose( average, np.zeros(d), rtol=0, atol=0.01 )
 

diff --git a/tests/test_symbolic.py b/tests/test_symbolic.py
@@ -24,7 +24,7 @@ def test_complex_expression(self):
 		self.assertTrue(has_function(expression,f))
 		self.assertEqual(
 				replace_function(expression,f,g),
-				3**g(42) + 23 - g(43,44) + g(45+a)*sin( g(g(46,47,48)+17) - g(4) + sin(g(42)) )
+				3**g(42) + 23 - g(43,44) + g(45+a)*sin( g(g(46,47,48)+17) - g(4) ) + sin(g(42))
 			)
 
 	def test_function_within_function(self):