Neo is now stricter about using mocks, the mock has to say what it is…

… pretending to be
NeuralEnsemble · Feb 20, 2024 · 83416b4 · 83416b4
1 parent c12a057
commit 83416b4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/unittests/test_recording.py b/test/unittests/test_recording.py
@@ -4,6 +4,7 @@
 from unittest.mock import Mock
 import pytest
 
+import neo
 from pyNN import recording, errors
 from pyNN.recording import Variable
 
@@ -60,6 +61,7 @@ class MockRecorder(recording.Recorder):
     def _get_current_segment(self, filter_ids=None, variables='all', clear=False):
         segment = Mock()
         segment.analogsignals = [Mock(), Mock()]
+        segment.proxy_for = neo.Segment
         return segment
 
     def _localize_variables(self, variables, locations):