From 890ee84634a22469e64f8a23279a40db8a22b5c2 Mon Sep 17 00:00:00 2001
From: Alexander Tischenko <sashaaero@yandex.ru>
Date: Wed, 3 Oct 2018 18:04:37 +0300
Subject: [PATCH] Fixes #385: Tests fail with python3.6

---
 setup.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 892beb2a..e1689e30 100644
--- a/setup.py
+++ b/setup.py
@@ -3,6 +3,13 @@
 from setuptools import setup
 import sys
 
+import unittest
+
+def test_suite():
+    test_loader = unittest.TestLoader()
+    test_suite = test_loader.discover('pony.orm.tests', pattern='test_*.py')
+    return test_suite
+
 name = "pony"
 version = __import__('pony').__version__
 description = "Pony Object-Relational Mapper"
@@ -116,5 +123,6 @@
         license=licence,
         packages=packages,
         package_data=package_data,
-        download_url=download_url
+        download_url=download_url,
+        test_suite='setup.test_suite'
     )