add othher new fields to tests

Author: samuelbray32

src/pynwb/tests/test_optogenetics.py

@@ -1,6 +1,7 @@
 import numpy as np
 from pynwb import NWBHDF5IO
 from pynwb.testing.mock.file import mock_NWBFile
+from pynwb.testing.mock.base import mock_TimeSeries
 from unittest import TestCase
 
 from ndx_franklab_novela import FrankLabOptogeneticEpochsTable, CameraDevice
@@ -11,6 +12,9 @@ class TestFrankLabOptogeneticsEpochsTable(TestCase):
     def test_roundtrip(self):
         nwbfile = mock_NWBFile()
 
+        stimulus = mock_TimeSeries()
+        nwbfile.add_stimulus(stimulus)
+
         camera1 = CameraDevice(
             name="overhead_run_camera 1",
             description="Camera used for tracking running",
@@ -68,7 +72,10 @@ def test_roundtrip(self):
             ripple_filter_lockout_period_in_samples=10,
             ripple_filter_threshold_sd=5.0,
             ripple_filter_num_above_threshold=4,
+            speed_filter_on=True,
+            speed_filter_threshold_in_cm_per_s=10.0,
             speed_filter_on_above_threshold=True,
+            stimulus_signal=stimulus,
         )
         nwbfile.add_time_intervals(opto_epochs)
 
@@ -114,3 +121,7 @@ def test_roundtrip(self):
             assert read_epochs[0, "ripple_filter_lockout_period_in_samples"] == 10
             assert read_epochs[0, "ripple_filter_threshold_sd"] == 5.0
             assert read_epochs[0, "ripple_filter_num_above_threshold"] == 4
+            assert read_epochs[0, "speed_filter_on"]
+            assert read_epochs[0, "speed_filter_threshold_in_cm_per_s"] == 10.0
+            assert read_epochs[0, "speed_filter_on_above_threshold"]
+            assert read_epochs[0, "stimulus_signal"].object_id == stimulus.object_id