From 29106473e3dcd08b96b9b1226ca948f838f34db6 Mon Sep 17 00:00:00 2001
From: Jannis Konrad <kabel42@gmail.com>
Date: Thu, 14 Sep 2017 20:30:56 +0200
Subject: [PATCH] fix tests

---
 DTLSSocket/__init__.py    | 1 +
 tests/testSocket.py       | 3 ++-
 tests/testSocketServer.py | 3 ++-
 3 files changed, 5 insertions(+), 2 deletions(-)
 create mode 100644 DTLSSocket/__init__.py

diff --git a/DTLSSocket/__init__.py b/DTLSSocket/__init__.py
new file mode 100644
index 0000000..cdbe217
--- /dev/null
+++ b/DTLSSocket/__init__.py
@@ -0,0 +1 @@
+from .DTLSSocket import DTLSSocket as DTLSSocket
diff --git a/tests/testSocket.py b/tests/testSocket.py
index f4ed18f..90d716f 100644
--- a/tests/testSocket.py
+++ b/tests/testSocket.py
@@ -1,4 +1,5 @@
-import dtls, time, socket
+from DTLSSocket import dtls
+import time, socket
 
 sock = socket.socket(family=socket.AF_INET6, type=socket.SOCK_DGRAM)
 sock.setsockopt(socket.IPPROTO_IPV6, socket.IPV6_RECVPKTINFO, 1)
diff --git a/tests/testSocketServer.py b/tests/testSocketServer.py
index cf57bbc..5ea2c94 100644
--- a/tests/testSocketServer.py
+++ b/tests/testSocketServer.py
@@ -1,4 +1,5 @@
-import dtls, time, socket
+from DTLSSocket import dtls
+import time, socket
 
 sock = socket.socket(family=socket.AF_INET6, type=socket.SOCK_DGRAM)
 sock.setsockopt(socket.IPPROTO_IPV6, socket.IPV6_RECVPKTINFO, 1)
-- 
GitLab