1 files changed, 20 insertions, 1 deletions
diff --git a/lab_control/test/jds6600_test.py b/lab_control/test/jds6600_test.py
index 817ccf4..59e2d33 100644
--- a/lab_control/test/jds6600_test.py
+++ b/lab_control/test/jds6600_test.py
@@ -17,7 +17,7 @@ def uut(mockDevice):
 
 def checkNumericalParameter(testValues, writeValue, valueInMock):
     for ch in JDS6600.AVAILABLE_CHANNELS:
-        assert valueInMock(ch) == None
+        assert valueInMock(ch) is None
 
         for value in testValues:
             writeValue(ch, value)
@@ -75,3 +75,22 @@ def test_invalidChannel(uut):
         with pytest.raises(AssertionError):
             uut.setOff(ch)
 
+def test_setFrequencySingleFailure(uut, mockDevice):
+    testFrequency = 1000.0
+    testChannel = 1
+    assert mockDevice.getFrequency(testChannel) is None
+
+    mockDevice.injectFailures(1)
+    uut.setFrequency(testChannel, testFrequency)
+
+    assert mockDevice.getFrequency(testChannel) == testFrequency
+
+def test_setFrequencyMultipleFailures(uut, mockDevice):
+    testFrequency = 1000.0
+    testChannel = 1
+    assert mockDevice.getFrequency(testChannel) is None
+
+    mockDevice.injectFailures(2)
+    uut.setFrequency(testChannel, testFrequency)
+
+    assert mockDevice.getFrequency(testChannel) == 0.0