python-periphery/tests/test_led.py at master · vsergeev/python-periphery · GitHub

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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
import os
import sys
import time

import periphery
from .test import ptest, pokay, passert, AssertRaises

if sys.version_info[0] == 3:
    raw_input = input


led_name = None


def test_arguments():
    ptest()

    # Invalid open types
    with AssertRaises("invalid open types", TypeError):
        periphery.LED("abc", "out")
    with AssertRaises("invalid open types", TypeError):
        periphery.LED(100, 100)


def test_open_close():
    ptest()

    # Open non-existent LED
    with AssertRaises("non-existent led", LookupError):
        periphery.LED("invalid_led_XXX", 0)

    # Open legitimate LED
    led = periphery.LED(led_name, 0)
    passert("property name", led.name == led_name)
    passert("fd >= 0", led.fd >= 0)
    passert("max_brightness > 0", led.max_brightness > 0)

    # Check triggers list
    triggers = led.triggers
    passert("triggers is a list", isinstance(triggers, list))
    passert("triggers count > 0", len(triggers) > 0)
    # Check each trigger entry
    passert("each trigger is a non-empty string", all(isinstance(t, str) and t for t in triggers))

    # Set brightness to True, check brightness
    led.write(True)
    time.sleep(0.01)
    passert("brightness is non-zero", led.read() > 0)

    # Set brightness to False, check brightness
    led.write(False)
    time.sleep(0.01)
    passert("brightness is zero", led.read() == 0)

    # Set brightness to 1, check brightness
    led.write(1)
    time.sleep(0.01)
    passert("brightness is non-zero", led.read() >= 1)

    # Set brightness to 0, check brightness
    led.write(0)
    time.sleep(0.01)
    passert("brightness is zero", led.read() == 0)

    # Set brightness to 1, check brightness
    led.brightness = 1
    time.sleep(0.01)
    passert("brightness is non-zero", led.brightness >= 1)

    # Set brightness to 0, check brightness
    led.brightness = 0
    time.sleep(0.01)
    passert("brightness is zero", led.brightness == 0)

    # Set trigger to default, check it isn't none
    led.trigger = "default"
    passert("trigger is not none", led.trigger != "none")

    # Set trigger to none, check it is none
    led.trigger = "none"
    passert("trigger is none", led.trigger == "none")

    # Set trigger to invalid type
    with AssertRaises("invalid trigger type", TypeError):
        led.trigger = 100

    led.close()


def test_loopback():
    ptest()
    # No general way to do a loopback test for I2C without a real component, skipping...


def test_interactive():
    ptest()

    led = periphery.LED(led_name, False)

    raw_input("Press enter to continue...")

    # Check tostring
    print("LED description: {}".format(str(led)))
    passert("interactive success", raw_input("LED description looks ok? y/n ") == "y")

    # Check triggers
    print("LED active trigger: {}".format(led.trigger))
    print("LED available triggers ({}): {}".format(len(led.triggers), " ".join(led.triggers)))
    passert("interactive success", raw_input("LED triggers look OK? y/n ") == "y")

    # Turn LED off
    led.write(False)
    passert("interactive success", raw_input("LED is off? y/n ") == "y")

    # Turn LED on
    led.write(True)
    passert("interactive success", raw_input("LED is on? y/n ") == "y")

    # Turn LED off
    led.write(False)
    passert("interactive success", raw_input("LED is off? y/n ") == "y")

    # Turn LED on
    led.write(True)
    passert("interactive success", raw_input("LED is on? y/n ") == "y")

    # Restore default trigger
    led.trigger = "default"

    led.close()


if __name__ == "__main__":
    if os.environ.get("CI") == "true":
        test_arguments()
        sys.exit(0)

    if len(sys.argv) < 2:
        print("Usage: python -m tests.test_led <LED name>")
        print("")
        print("[1/4] Arguments test: No requirements.")
        print("[2/4] Open/close test: LED should be real.")
        print("[3/4] Loopback test: No test.")
        print("[4/4] Interactive test: LED should be observed.")
        print("")
        print("Hint: for Raspberry Pi 3, disable triggers for led1:")
        print("    $ echo none > /sys/class/leds/led1/trigger")
        print("Observe led1 (red power LED), and run this test:")
        print("    python -m tests.test_led led1")
        print("")
        sys.exit(1)

    led_name = sys.argv[1]

    test_arguments()
    pokay("Arguments test passed.")
    test_open_close()
    pokay("Open/close test passed.")
    test_loopback()
    pokay("Loopback test passed.")
    test_interactive()
    pokay("Interactive test passed.")

    pokay("All tests passed!")