1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
import pytest
from lab_control.jds6600 import JDS6600
from lab_control.test.mock_jds6600_device import MockJDS6600Device
@pytest.fixture
def mockDevice():
d = MockJDS6600Device()
yield d
d.stop()
@pytest.fixture
def uut(mockDevice):
uut = JDS6600(mockDevice.getPortName())
yield uut
uut.closePort()
def checkNumericalParameter(testValues, writeValue, valueInMock):
for ch in JDS6600.AVAILABLE_CHANNELS:
assert valueInMock(ch) is None
for value in testValues:
writeValue(ch, value)
assert valueInMock(ch) == value
def checkInvalidNumericalParameter(testValues, writeValue, valueInMock):
for ch in JDS6600.AVAILABLE_CHANNELS:
for value in testValues:
with pytest.raises(AssertionError):
writeValue(ch, value)
def test_serialConfiguration(mockDevice):
with pytest.raises(AssertionError):
mockDevice.checkPortConfiguration()
uut = JDS6600(mockDevice.getPortName())
mockDevice.checkPortConfiguration()
def test_channelOnAndOff(uut, mockDevice):
for ch in JDS6600.AVAILABLE_CHANNELS:
assert not mockDevice.isOn(ch)
uut.setOn(ch)
assert mockDevice.isOn(ch)
uut.setOff(ch)
assert not mockDevice.isOn(ch)
def test_setFrequency(uut, mockDevice):
checkNumericalParameter([0.0, 100.0, 100000.0, 60000000.0], uut.setFrequency, mockDevice.getFrequency)
def test_setInvalidFrequency(uut, mockDevice):
checkInvalidNumericalParameter([-10.0, 60000000.1, None], uut.setFrequency, mockDevice.getFrequency)
def test_setAmplitude(uut, mockDevice):
checkNumericalParameter([0.0, 0.1, 1.0, 10.0, 20.0], uut.setAmplitude, mockDevice.getAmplitude)
def test_setInvalidAmplitude(uut, mockDevice):
checkInvalidNumericalParameter([-0.1, -10.0, 20.1, None], uut.setAmplitude, mockDevice.getAmplitude)
def test_setFunction(uut, mockDevice):
checkNumericalParameter(range(0, 17), uut.setFunction, mockDevice.getFunction)
def test_setInvalidFunction(uut, mockDevice):
checkInvalidNumericalParameter([-1, -10, 17, 20, None], uut.setFunction, mockDevice.getFunction)
def test_invalidChannel(uut):
testMethods = [uut.setFrequency, uut.setAmplitude, uut.setFunction]
for ch in [-1, 0, 3, None]:
for method in testMethods:
with pytest.raises(AssertionError):
method(ch, 0)
with pytest.raises(AssertionError):
uut.setOn(ch)
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
|
