Fix test to catch correct exception for latest numpy version

brian-team · Jun 24, 2021 · 62fa449 · 62fa449
1 parent d71837c
commit 62fa449
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/brian2/tests/test_units.py b/brian2/tests/test_units.py
@@ -978,7 +978,7 @@ def test_numpy_functions_logical():
                 # one argument
                 result_units = eval('np.%s(value1)' % ufunc)        
                 result_array = eval('np.%s(np.array(value1))' % ufunc)                
-            except ValueError:
+            except (ValueError, TypeError):
                 # two arguments
                 result_units = eval('np.%s(value1, value2)' % ufunc)        
                 result_array = eval('np.%s(np.array(value1), np.array(value2))' % ufunc)