Handle no Reference Electrode (#127)

* handle no ref case

* add test for no ref case

* Update src/trodes_to_nwb/tests/test_convert_yaml.py

Co-authored-by: Copilot <[email protected]>

* update test order in file for readability

---------

Co-authored-by: Copilot <[email protected]>

Author: samuelbray32

src/trodes_to_nwb/convert_yaml.py

@@ -300,6 +300,10 @@ def add_electrode_groups(
     for nwb_group in list(electrode_table["group_name"]):
         # use the refference electrode map and defined electrode table to find index of the reference electrode
         ref_group, ref_electrode = ref_electrode_map[str(nwb_group)]
+        if ref_group == "-1":
+            # handle case where no reference electrode is defined
+            ref_electrode_id.append(-1)
+            continue
         ref_electrode_id.append(
             electrode_table.index[
                 (electrode_table["group_name"] == ref_group)

src/trodes_to_nwb/tests/test_convert_yaml.py

@@ -236,6 +236,31 @@ def test_electrode_creation_reconfigured():
     assert list(nwbfile.electrodes.to_dataframe()["hwChan"][-2:]) == ["115", "107"]
 
 
+def test_no_reference_electrode():
+    metadata_path = data_path / "20230622_sample_metadata.yml"
+    probe_metadata = [data_path / "tetrode_12.5.yml"]
+    metadata, probe_metadata = convert_yaml.load_metadata(metadata_path, probe_metadata)
+    nwbfile = convert_yaml.initialize_nwb(metadata, default_test_xml_tree())
+
+    # create the hw_channel map using rec data
+    recfile = data_path / "20230622_sample_01_a1.rec"
+    rec_header = convert_rec_header.read_header(recfile)
+    hw_channel_map = convert_rec_header.make_hw_channel_map(
+        metadata, rec_header.find("SpikeConfiguration")
+    )
+    ref_electrode_map = convert_rec_header.make_ref_electrode_map(
+        metadata, rec_header.find("SpikeConfiguration")
+    )
+    # set one tetrode to no reference
+    ref_electrode_map["0"] = ("-1", -1)
+    convert_yaml.add_electrode_groups(
+        nwbfile, metadata, probe_metadata, hw_channel_map, ref_electrode_map
+    )
+    df = nwbfile.electrodes.to_dataframe()
+    assert (df[df["group_name"] == "0"].ref_elect_id.unique() == -1).all()
+    assert (df[df["group_name"] != "0"].ref_elect_id.unique() == 0).all()
+
+
 def test_add_tasks():
     # Set up test data
     metadata_path = data_path / "20230622_sample_metadata.yml"