1
0
mirror of https://github.com/cmur2/python-veml6070.git synced 2024-12-21 14:54:23 +01:00

module: add pytest for unit and snapshot testing with travis-ci

This commit is contained in:
cn 2018-03-14 00:39:10 +01:00
parent 8cfb2541b9
commit 2086f111f2
9 changed files with 244 additions and 1 deletions

3
.gitignore vendored
View File

@ -1 +1,4 @@
.cache/
env/
__pycache__/
*.pyc

8
.travis.yml Normal file
View File

@ -0,0 +1,8 @@
---
sudo: false
language: python
python:
- "2.7"
script:
- pytest

View File

@ -1,6 +1,7 @@
# python-veml6070
[![Build Status](https://travis-ci.org/cmur2/python-veml6070.svg?branch=master)](https://travis-ci.org/cmur2/python-veml6070)
A Python library for accessing the [VEML6070 digital UV light sensor](http://www.vishay.com/docs/84277/veml6070.pdf) from Vishay via `python-smbus` using the I2C interface.
Default settings are suitable for Raspberry Pi 2 and 3 and was successfully tested using a [breakout](https://github.com/watterott/VEML6070-Breakout).

2
pytest.ini Normal file
View File

@ -0,0 +1,2 @@
[pytest]
norecursedirs = env* .*

4
requirements.txt Normal file
View File

@ -0,0 +1,4 @@
mock==2.0.0
pylint==1.8.2
pytest==3.0.6
snapshottest==0.5.0

0
tests/__init__.py Normal file
View File

View File

View File

@ -0,0 +1,128 @@
# -*- coding: utf-8 -*-
# snapshottest: v1 - https://goo.gl/zC4yUc
from __future__ import unicode_literals
from snapshottest import Snapshot
snapshots = Snapshot()
snapshots['TestVeml6070::test_disable 1'] = {
'readlog': [
],
'writelog': [
(
'56',
[
6,
6,
6
]
)
]
}
snapshots['TestVeml6070::test_enable 1'] = {
'readlog': [
],
'writelog': [
(
'56',
[
6,
6,
6
]
)
]
}
snapshots['TestVeml6070::test_integration_time 1'] = {
'readlog': [
],
'writelog': [
(
'56',
[
2,
2,
14
]
)
]
}
snapshots['TestVeml6070::test_setup 1'] = {
'readlog': [
],
'writelog': [
(
'56',
[
6,
6
]
)
]
}
snapshots['TestVeml6070::test_uva_light_intensity 1'] = {
'readlog': [
(
'56',
[
6,
6
]
),
(
'57',
[
1,
1
]
)
],
'writelog': [
(
'56',
[
6,
6,
6,
6,
14,
14,
14
]
)
]
}
snapshots['TestVeml6070::test_uva_light_intensity_raw 1'] = {
'readlog': [
(
'56',
[
52
]
),
(
'57',
[
18
]
)
],
'writelog': [
(
'56',
[
6,
6,
6,
6
]
)
]
}

97
tests/test_veml6070.py Normal file
View File

@ -0,0 +1,97 @@
# 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
# inspired by https://github.com/adafruit/Adafruit_Python_GPIO/blob/master/Adafruit_GPIO/I2C.py
class MockSMBus(object):
def __init__(self, initial_read=None):
self._writelog = {}
self._readlog = {}
self.initial_read = initial_read or {}
def read_byte(self, addr):
val = self.initial_read.get(addr).pop(0)
self._readlog.setdefault(addr, []).append(val)
return val
def write_byte(self, addr, val):
self._writelog.setdefault(addr, []).append(val)
def _get_log(self):
return {
'readlog': [(str(key), value) for (key, value) in self._readlog.items()],
'writelog': [(str(key), value) for (key, value) in self._writelog.items()]
}
# 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
class TestVeml6070(snapshottest.TestCase):
def test_setup(self):
mockbus = setup_mockbus()
veml = veml6070.Veml6070()
self.assertIsNotNone(veml)
self.assertMatchSnapshot(mockbus._get_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._get_log())
def test_enable(self):
mockbus = setup_mockbus()
veml = veml6070.Veml6070()
veml.enable()
self.assertMatchSnapshot(mockbus._get_log())
def test_disable(self):
mockbus = setup_mockbus()
veml = veml6070.Veml6070()
veml.disable()
self.assertMatchSnapshot(mockbus._get_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._get_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.05625 / 1)
veml.set_integration_time(veml6070.INTEGRATIONTIME_4T)
self.assertEqual(veml.get_uva_light_intensity(), 0x0106 * 0.05625 / 4)
self.assertMatchSnapshot(mockbus._get_log())