Configure pytest

.editorconfig

@@ -16,3 +16,6 @@ indent_style = space
 
 [*.{yaml,yml,md}]
 indent_size = 2
+
+[.vscode/*.json]
+insert_final_newline = false

.vscode/settings.json

@@ -1,6 +1,8 @@
 {
-	"python.linting.mypyEnabled": true,
-	"python.formatting.provider": "black",
 	"editor.formatOnSave": true,
-	"python.linting.flake8Enabled": true
+	"pylint.args": [
+		"--disable=missing-function-docstring,missing-class-docstring,missing-module-docstring"
+	],
+	"python.testing.unittestEnabled": false,
+	"python.testing.pytestEnabled": true
 }

misc/render_rrd.py

@@ -5,7 +5,7 @@ from ast import literal_eval
 from collections import namedtuple
 from typing import Any, Dict
 
-import rrdtool
+import rrdtool  # type: ignore
 
 from srnemqtt.solar_types import DataName
 
@@ -147,7 +147,6 @@ def rrdupdate(file: str, timestamp: int, data: dict):
 
 
 def re_read():
-
     rrdtool.create(
         RRDFILE,
         # "--no-overwrite",

misc/test_bleuart.py

@@ -4,7 +4,6 @@ from srnemqtt.lib.feasycom_ble import BTLEUart
 from srnemqtt.protocol import construct_request, write
 
 with BTLEUart(MAC, timeout=1) as x:
-
     print(x)
 
     write(x, construct_request(0x0E, words=3))

misc/test_serial.py

@@ -3,7 +3,7 @@ import os
 import sys
 from time import sleep
 
-from serial import Serial
+from serial import Serial  # type: ignore
 
 print(sys.path)
 sys.path.insert(1, os.path.dirname(os.path.dirname(sys.argv[0])))

misc/test_serial_loopback.py

@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
-from serial import Serial
+from serial import Serial  # type: ignore
 
 with Serial("/dev/ttyUSB0", baudrate=9600, timeout=2) as x:
     x.write(b"Hello, World!")

srnemqtt/__main__.py

@@ -5,8 +5,8 @@ import time
 from decimal import Decimal
 from typing import cast
 
-from bluepy.btle import BTLEDisconnectError
+from bluepy.btle import BTLEDisconnectError  # type: ignore
-from serial import SerialException
+from serial import SerialException  # type: ignore
 
 from .config import get_config, get_consumers, get_interface
 from .protocol import parse_battery_state, parse_historical_entry, try_read_parse

srnemqtt/interfaces/serial.py

@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
-import serial
+import serial  # type: ignore
 
 from . import BaseInterface
 

srnemqtt/lib/feasycom_ble.py

@@ -4,7 +4,7 @@ import queue
 import time
 from typing import TYPE_CHECKING, Optional, cast
 
-from bluepy import btle
+from bluepy import btle  # type: ignore
 
 if TYPE_CHECKING:
     from _typeshed import ReadableBuffer, WriteableBuffer

srnemqtt/protocol.py

@@ -5,7 +5,7 @@ import time
 from io import RawIOBase
 from typing import Callable, Collection, Optional
 
-from libscrc import modbus
+from libscrc import modbus  # type: ignore
 
 from .constants import ACTION_READ, POSSIBLE_MARKER
 from .lib.feasycom_ble import BTLEUart
@@ -84,7 +84,6 @@ def discardUntil(fh: RawIOBase, byte: int, timeout=10) -> Optional[int]:
     discarded = 0
     read_byte = expand(fh.read(1))
     while read_byte != byte:
-
         if read_byte is not None:
             if not discarded:
                 log("Discarding", end="")

tests/test_protocol.py

@@ -0,0 +1,11 @@
+from io import BytesIO
+
+from srnemqtt.protocol import write as protocol_write
+
+
+def test_write():
+    fh = BytesIO()
+    protocol_write(fh, b"Hello, World!")
+    fh.seek(0)
+
+    assert fh.read() == b"Hello, World!\x4E\x11"

tox.ini

@@ -1,3 +1,7 @@
 [flake8]
 max-line-length = 88
 extend-ignore = E203, I201, I101
+
+[pytest]
+pythonpath = .
+testpaths = tests