just switch to run

Author: Cody Baker

tests/unit_tests/test_tables.py

@@ -1,6 +1,5 @@
 import platform
 import json
-import pytest
 from unittest import TestCase
 from packaging import version
 
@@ -229,28 +228,34 @@ def test_check_single_row_pass():
 
 
 def test_check_single_row_ignore_units():
-    table = Units(
-        name="Units",
-    )  # default name when building through nwbfile
+    table = Units(name="Units",)  # default name when building through nwbfile
     table.add_unit(spike_times=[1, 2, 3])
     assert check_single_row(table=table) is None
 
 
-@pytest.mark.skipif(get_package_version(name="pynwb") >= version.Version("2.1.0"))  # TODO: remove skip when fixed
 def test_check_single_row_ignore_electrodes():
-    table = ElectrodeTable(
-        name="electrodes",
-    )  # default name when building through nwbfile
-    table.add_row(
-        x=np.nan,
-        y=np.nan,
-        z=np.nan,
-        imp=np.nan,
-        location="unknown",
-        filtering="unknown",
-        group=ElectrodeGroup(name="test_group", description="", device=Device(name="test_device"), location="unknown"),
-        group_name="test_group",
-    )
+    table = ElectrodeTable(name="electrodes",)  # default name when building through nwbfile
+    if get_package_version(name="pynwb") >= version.Version("2.1.0"):
+        table.add_row(
+            location="unknown",
+            group=ElectrodeGroup(
+                name="test_group", description="", device=Device(name="test_device"), location="unknown"
+            ),
+            group_name="test_group",
+        )
+    else:
+        table.add_row(
+            x=np.nan,
+            y=np.nan,
+            z=np.nan,
+            imp=np.nan,
+            location="unknown",
+            filtering="unknown",
+            group=ElectrodeGroup(
+                name="test_group", description="", device=Device(name="test_device"), location="unknown"
+            ),
+            group_name="test_group",
+        )
     assert check_single_row(table=table) is None