Added missing unit tests

Author: MichaelClerx

syncropatch_export/trace.py

@@ -33,10 +33,9 @@ class Trace:
     def __init__(self, filepath, json_file: str):
         # store file paths
         self.filepath = filepath
-        if json_file[-5:] == '.json':
-            self.json_file = json_file
-        else:
-            self.json_file = json_file + '.json'
+        self.json_file = json_file
+        if not json_file.endswith('.json'):
+            self.json_file += '.json'
 
         # load json file
         with open(os.path.join(self.filepath, self.json_file)) as f:

tests/test_trace_class.py

@@ -41,6 +41,12 @@ def test_protocol_descriptions(self):
         self.assertLess(t_error, 1e-2)
         self.assertLess(v_error, 1e-4)
 
+    def test_get_protocol_description(self):
+        a = np.array(self.trace.get_protocol_description())
+        b = np.array(self.trace.get_voltage_protocol().get_all_sections())
+        self.assertEqual(a.shape, b.shape)
+        self.assertTrue(np.all(a == b))
+
     def test_protocol_export(self):
         with tempfile.TemporaryDirectory() as d:
             protocol = self.trace.get_voltage_protocol()