mirror of
https://github.com/cmur2/python-veml6070.git
synced 2024-12-22 10:54:23 +01:00
113 lines
4.3 KiB
Python
113 lines
4.3 KiB
Python
# import unittest
|
|
import sys
|
|
import time
|
|
|
|
import mock
|
|
|
|
import snapshottest
|
|
|
|
# Note: prepare globally mocked modules first and then load our module
|
|
MOCKED_SMBUS_MODULE = mock.Mock()
|
|
sys.modules['smbus'] = MOCKED_SMBUS_MODULE
|
|
time.sleep = lambda s: None
|
|
import veml6070 # pylint: disable=wrong-import-position
|
|
|
|
|
|
# inspired by https://github.com/adafruit/Adafruit_Python_GPIO/blob/master/Adafruit_GPIO/I2C.py
|
|
class MockSMBus(object): # pylint: disable=bad-option-value,useless-object-inheritance
|
|
def __init__(self, initial_read=None):
|
|
self._log = []
|
|
self.initial_read = initial_read or {}
|
|
|
|
def read_byte(self, addr):
|
|
val = self.initial_read.get(addr).pop(0)
|
|
self._log.append(('r', addr, val))
|
|
return val
|
|
|
|
def write_byte(self, addr, val):
|
|
self._log.append(('w', addr, val))
|
|
|
|
|
|
# def create_veml6070(**kwargs):
|
|
# mockbus = MockSMBus()
|
|
# smbus = Mock()
|
|
# smbus.SMBus.return_value = mockbus
|
|
# with patch.dict('sys.modules', {'smbus': smbus}):
|
|
# import veml6070
|
|
# # Note: our module constants only available in this scope
|
|
# return (veml6070.Veml6070(**kwargs), mockbus)
|
|
|
|
|
|
def setup_mockbus(**kwargs):
|
|
mockbus = MockSMBus(**kwargs)
|
|
MOCKED_SMBUS_MODULE.SMBus.return_value = mockbus
|
|
return mockbus
|
|
|
|
|
|
# pylint: disable=protected-access
|
|
class TestVeml6070(snapshottest.TestCase):
|
|
def test_setup(self):
|
|
mockbus = setup_mockbus()
|
|
veml = veml6070.Veml6070()
|
|
MOCKED_SMBUS_MODULE.SMBus.assert_called_with(1)
|
|
self.assertIsNotNone(veml)
|
|
self.assertMatchSnapshot(mockbus._log)
|
|
|
|
def test_integration_time(self):
|
|
mockbus = setup_mockbus()
|
|
veml = veml6070.Veml6070(integration_time=veml6070.INTEGRATIONTIME_1_2T)
|
|
self.assertEqual(veml.get_integration_time(), veml6070.INTEGRATIONTIME_1_2T)
|
|
veml.set_integration_time(veml6070.INTEGRATIONTIME_4T)
|
|
self.assertEqual(veml.get_integration_time(), veml6070.INTEGRATIONTIME_4T)
|
|
self.assertMatchSnapshot(mockbus._log)
|
|
|
|
def test_enable(self):
|
|
mockbus = setup_mockbus()
|
|
veml = veml6070.Veml6070()
|
|
veml.enable()
|
|
self.assertMatchSnapshot(mockbus._log)
|
|
|
|
def test_disable(self):
|
|
mockbus = setup_mockbus()
|
|
veml = veml6070.Veml6070()
|
|
veml.disable()
|
|
self.assertMatchSnapshot(mockbus._log)
|
|
|
|
def test_uva_light_intensity_raw(self):
|
|
mockbus = setup_mockbus(initial_read={0x38 + 1: [0x12], 0x38 + 0: [0x34]})
|
|
veml = veml6070.Veml6070()
|
|
self.assertEqual(veml.get_uva_light_intensity_raw(), 0x1234)
|
|
self.assertMatchSnapshot(mockbus._log)
|
|
|
|
def test_uva_light_intensity(self):
|
|
mockbus = setup_mockbus(initial_read={0x38 + 1: [0x01, 0x01], 0x38 + 0: [0x06, 0x06]})
|
|
veml = veml6070.Veml6070()
|
|
self.assertEqual(veml.get_uva_light_intensity(), 0x0106 * (0.1 / 0.1125) * 0.05 / 1)
|
|
veml.set_integration_time(veml6070.INTEGRATIONTIME_4T)
|
|
self.assertEqual(veml.get_uva_light_intensity(), 0x0106 * (0.1 / 0.1125) * 0.05 / 4)
|
|
self.assertMatchSnapshot(mockbus._log)
|
|
|
|
def test_get_estimated_risk_level(self):
|
|
mockbus = setup_mockbus(initial_read={0x38 + 1: [0x01, 0x04], 0x38 + 0: [0x06, 0x01]}) # pylint: disable=unused-variable
|
|
veml = veml6070.Veml6070()
|
|
intensity = veml.get_uva_light_intensity()
|
|
self.assertEqual(veml.get_estimated_risk_level(intensity), "low")
|
|
intensity = veml.get_uva_light_intensity()
|
|
self.assertEqual(veml.get_estimated_risk_level(intensity), "moderate")
|
|
# don't assert mockbus._log as we do not care
|
|
|
|
def test_get_refresh_time(self):
|
|
mockbus = setup_mockbus() # pylint: disable=unused-variable
|
|
veml = veml6070.Veml6070(rset=veml6070.RSET_240K)
|
|
self.assertEqual(veml.get_refresh_time(), 0.1)
|
|
veml.set_integration_time(veml6070.INTEGRATIONTIME_1_2T)
|
|
self.assertEqual(veml.get_refresh_time(), 0.05)
|
|
veml2 = veml6070.Veml6070(rset=veml6070.RSET_270K)
|
|
self.assertEqual(veml2.get_refresh_time(), 0.1125)
|
|
veml3 = veml6070.Veml6070(rset=veml6070.RSET_600K)
|
|
veml3.set_integration_time(veml6070.INTEGRATIONTIME_2T)
|
|
self.assertEqual(veml3.get_refresh_time(), 0.5)
|
|
veml4 = veml6070.Veml6070(rset=480000)
|
|
self.assertEqual(veml4.get_refresh_time(), 0.2)
|
|
# don't assert mockbus._log as we do not care
|