summaryrefslogtreecommitdiffstats
path: root/lab_control/test/jds6600_test.py
blob: 948b5fc52a9d1b885c20b39a22986416f35e25ae (plain)
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
import pytest

from lab_control.jds6600 import JDS6600
from lab_control.test.mock_jds6600_device import MockJDS6600Device

AVAILABLE_CHANNELS = [1, 2]

@pytest.fixture
def mockDevice():
    d = MockJDS6600Device()
    yield d
    d.stop()

@pytest.fixture
def uut(mockDevice):
    uut = JDS6600(mockDevice.getPortName())
    yield uut
    uut.closePort()

def test_serialConfiguration(mockDevice):
    with pytest.raises(AssertionError):
        mockDevice.checkPortConfiguration()

    uut = JDS6600(mockDevice.getPortName())
    mockDevice.checkPortConfiguration()

def test_channelOnAndOff(uut, mockDevice):
    for ch in 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):
    for ch in AVAILABLE_CHANNELS:
        assert mockDevice.getFrequency(ch) == 0.0

        for frequency in [100.0, 100000.0, 0.0]:
            uut.setFrequency(ch, frequency)
            assert mockDevice.getFrequency(ch) == frequency

def test_setInvalidFrequency(uut, mockDevice):
    for ch in AVAILABLE_CHANNELS:
        uut.setFrequency(ch, None)
        assert mockDevice.getFrequency(ch) == 0.0

        uut.setFrequency(ch, -10.0)
        assert mockDevice.getFrequency(ch) == 0.0

        uut.setFrequency(ch, 60000000.1)
        assert mockDevice.getFrequency(ch) == 0.0