Update unit test #360

Author: peterhollender

tests/test_protocol.py

@@ -46,12 +46,16 @@ def test_default_protocol():
 
 def test_to_table(example_protocol: Protocol):
     """Ensure that the protocol can be correctly converted to a table."""
-    df = example_protocol.to_table()
-    assert df is not None
-    assert "Protocol Parameter" in df.columns
-    assert "Value" in df.columns
-    assert "Units" in df.columns
-    assert df[df["Protocol Parameter"] == "ID"]["Value"].iloc[0] == "example_protocol"
+    t = example_protocol.to_table()
+    assert t is not None
+    assert "Category" in t.columns
+    assert "Name" in t.columns
+    assert "Value" in t.columns
+    assert "Unit" in t.columns
+    tm =t.set_index(["Category", "Name"])
+    assert tm.loc["","ID"]["Value"] == example_protocol.id
+    assert tm.loc["Delay Method", "Default Sound Speed"]["Value"] == 1540.0
+    assert tm.loc["Delay Method", "Default Sound Speed"]["Unit"] == "m/s"
 
 @pytest.mark.parametrize(
     "target_constraints",