Files
grow-python/library/tests/test_setup.py
Phil Howard 99ba73dc9f Update tests
2020-10-20 15:41:31 +01:00

48 lines
1.3 KiB
Python

import mock
def test_moisture_setup(GPIO, smbus):
from grow.moisture import Moisture
ch1 = Moisture(channel=1)
ch2 = Moisture(channel=2)
ch3 = Moisture(channel=3)
assert GPIO.setup.has_calls([
mock.call(ch1._gpio_pin, GPIO.IN),
mock.call(ch2._gpio_pin, GPIO.IN),
mock.call(ch3._gpio_pin, GPIO.IN)
])
def test_moisture_read(GPIO, smbus):
from grow.moisture import Moisture
assert Moisture(channel=1).saturation == 1.0
assert Moisture(channel=2).saturation == 1.0
assert Moisture(channel=3).saturation == 1.0
assert Moisture(channel=1).moisture == 0
assert Moisture(channel=2).moisture == 0
assert Moisture(channel=3).moisture == 0
def test_pump_setup(GPIO, smbus):
from grow.pump import Pump, PUMP_PWM_FREQ
ch1 = Pump(channel=1)
ch2 = Pump(channel=2)
ch3 = Pump(channel=3)
assert GPIO.setup.has_calls([
mock.call(ch1._gpio_pin, GPIO.OUT, initial=GPIO.LOW),
mock.call(ch2._gpio_pin, GPIO.OUT, initial=GPIO.LOW),
mock.call(ch3._gpio_pin, GPIO.OUT, initial=GPIO.LOW)
])
assert GPIO.PWM.has_calls([
mock.call(ch1._gpio_pin, PUMP_PWM_FREQ),
mock.call(ch2._gpio_pin, PUMP_PWM_FREQ),
mock.call(ch3._gpio_pin, PUMP_PWM_FREQ)
])