Full coverage tests for Minghe MHS5200a

instrumentkit · Nov 3, 2020 · 705e056 · 705e056
1 parent f2da132
commit 705e056
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/instruments/tests/test_minghe/test_minghe_mhs5200a.py b/instruments/tests/test_minghe/test_minghe_mhs5200a.py
@@ -207,3 +207,31 @@ def test_mhs_serial_number():
             sep="\r\n"
     ) as mhs:
         assert mhs.serial_number == "5225A1"
+
+
+def test_mhs_get_amplitude():
+    """Raise NotImplementedError when trying to get amplitude"""
+    with expected_protocol(
+            ik.minghe.MHS5200,
+            [
+            ],
+            [
+            ],
+            sep="\r\n"
+    ) as mhs:
+        with pytest.raises(NotImplementedError):
+            mhs._get_amplitude_()
+
+
+def test_mhs_set_amplitude():
+    """Raise NotImplementedError when trying to set amplitude"""
+    with expected_protocol(
+            ik.minghe.MHS5200,
+            [
+            ],
+            [
+            ],
+            sep="\r\n"
+    ) as mhs:
+        with pytest.raises(NotImplementedError):
+            mhs._set_amplitude_(1, 2)