cn/python-veml6070
1
0
Fork 0
mirror of https://github.com/cmur2/python-veml6070.git synced 2025-06-26 12:30:19 +02:00
Code Releases Activity

module: use yapf formatter, reduce pylint warnings

Browse Source
This commit is contained in:
cn
2020-08-07 23:01:51 +02:00
parent ded2908236
commit b358ea539f
7 changed files with 43 additions and 581 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
tests/veml6070_test.py
View File

@ -1,4 +1,3 @@
# import unittest
import sys
import time
@ -11,7 +10,8 @@ import snapshottest
MOCKED_SMBUS_MODULE = mock.Mock()
sys.modules['smbus'] = MOCKED_SMBUS_MODULE
time.sleep = lambda s: None
import veml6070
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
@ -27,6 +27,7 @@ class MockSMBus(object): # pylint: disable=bad-option-value,useless-object-inher
def write_byte(self, addr, val):
self._log.append(('w', addr, val))
# def create_veml6070(**kwargs):
# mockbus = MockSMBus()
# smbus = Mock()
@ -36,13 +37,15 @@ class MockSMBus(object): # pylint: disable=bad-option-value,useless-object-inher
# # 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
class TestVeml6070(snapshottest.TestCase):
# pylint: disable=protected-access
class TestVeml6070(snapshottest.TestCase):
def test_setup(self):
mockbus = setup_mockbus()
veml = veml6070.Veml6070()
@ -71,38 +74,30 @@ class TestVeml6070(snapshottest.TestCase):
self.assertMatchSnapshot(mockbus._log)
def test_uva_light_intensity_raw(self):
mockbus = setup_mockbus(initial_read={
0x38+1: [0x12],
0x38+0: [0x34]
})
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]
})
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)
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.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]
})
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()
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)