From 24219b38f9e6dfb18c439573e241377690667028 Mon Sep 17 00:00:00 2001
From: Christian Decker <decker.christian@gmail.com>
Date: Fri, 31 Jan 2020 20:45:50 +0100
Subject: [PATCH] pyln: Add unit tests from pyln-* to `make check-python`

---
 Makefile                                  | 3 ++-
 contrib/pylightning/lightning/__init__.py | 9 +++++++++
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/Makefile b/Makefile
index 3a1cea5b3..0975ca4c8 100644
--- a/Makefile
+++ b/Makefile
@@ -314,9 +314,10 @@ check-python:
 	@# E501 line too long (N > 79 characters)
 	@# E731 do not assign a lambda expression, use a def
 	@# W503: line break before binary operator
-	@flake8 --ignore=E501,E731,W503 --exclude=contrib/pylightning/lightning/__init__.py ${PYSRC}
+	@flake8 --ignore=E501,E731,W503 ${PYSRC}
 
 	PYTHONPATH=contrib/pyln-client:$$PYTHONPATH $(PYTEST) contrib/pyln-client/
+	PYTHONPATH=contrib/pyln-proto:$$PYTHONPATH $(PYTEST) contrib/pyln-proto/
 
 check-includes:
 	@tools/check-includes.sh
diff --git a/contrib/pylightning/lightning/__init__.py b/contrib/pylightning/lightning/__init__.py
index 2836c5e16..a24910b21 100644
--- a/contrib/pylightning/lightning/__init__.py
+++ b/contrib/pylightning/lightning/__init__.py
@@ -1 +1,10 @@
 from pyln.client import LightningRpc, RpcError, Millisatoshi, __version__, Plugin, monkey_patch
+
+__all__ = [
+    "__version__",
+    "LightningRpc",
+    "RpcError",
+    "Millisatoshi",
+    "Plugin",
+    "monkey_patch",
+]