A python library for accessing the VEML6070 digital UV light sensor from Vishay
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 

112 lines
4.3 KiB

# 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