Set up testing with pytest

.gitignore

@@ -42,3 +42,4 @@ docs/*.pdf
 
 # Python
 VIRTUAL/
+.cache/

.travis.yml

@@ -1,8 +1,11 @@
 language: python
 python:
-    - 2.7
-    - 3.4
+  - 2.7
+  - 3.2
+  - 3.3
+  - 3.4
+  - 3.5
 install:
-    - pip install flake8
+    - pip install -r requirements-dev.txt
 script:
-    - flake8 --max-line-length=99 --ignore=E126,E127,E128,C901 RPLCD/[a-zA-Z]*.py
+    - py.test -v

README.rst

@@ -88,6 +88,9 @@ You can find the documentation here: https://readthedocs.org/projects/rplcd/
 Testing
 =======
 
+Test Scripts
+------------
+
 To test your 20x4 display, please run the ``test_20x4.py`` script and
 confirm/verify each step with the enter key. If you don't use the standard
 wiring, make sure to add your pin numbers to the ``CharLCD`` constructor in
@@ -96,6 +99,17 @@ wiring, make sure to add your pin numbers to the ``CharLCD`` constructor in
 To test a 16x2 display, proceed as explained above, but use the ``test_16x2.py``
 script instead.
 
+Unit Tests
+----------
+
+There are also unit tests. First, install dependencies:
+
+    pip install -U -r requirements-dev.txt
+
+Then run the tests:
+
+    py.test -v
+
 
 Coding Guidelines
 =================

RPLCD/__init__.py

@@ -4,6 +4,7 @@
 from .contextmanagers import cursor, cleared
 from .gpio import CharLCD as GpioCharLCD
 
+
 class CharLCD(GpioCharLCD):
     def __init__(self, *args, **kwargs):
         warnings.warn("Using RPLCD.CharLCD directly is deprecated. " +

pytest.ini

@@ -0,0 +1,12 @@
+[pytest]
+addopts = --pep8 --tb=short
+python_paths = ..
+python_files = tests/test_*.py
+pep8ignore =
+    *.py E126 E127 E128
+    */tests/* ALL
+    docs/* ALL
+    setup.py ALL
+    test_16x2.py ALL
+    test_20x4.py ALL
+pep8maxlinelength = 99

requirements-dev.txt

@@ -0,0 +1,4 @@
+pytest==3.0.5
+pytest-mock==1.5.0
+pytest-pep8==1.0.6
+pytest-pythonpath==0.7.1

tests/conftest.py

@@ -0,0 +1,11 @@
+import mock
+
+
+# Mock RPi.GPIO module (https://m.reddit.com/r/Python/comments/5eddp5/mock_testing_rpigpio/)
+MockRPi = mock.MagicMock()
+modules = {
+    'RPi': MockRPi,
+    'RPi.GPIO': MockRPi.GPIO,
+}
+patcher = mock.patch.dict('sys.modules', modules)
+patcher.start()