Add working test setup

1 files changed, 16 insertions, 2 deletions

diff --git a/setup.py b/setup.py
index 23152cc4..7fdb70f1 100644
--- a/setup.py
+++ b/setup.py
@@ -24,8 +24,22 @@ setup(
     ],
     keywords='opendc datacenter simulation web-server',
     packages=['opendc'],
+    # yapf: disable
     install_requires=[
-        'flask==1.0.2', 'flask-socketio==3.0.2', 'oauth2client==4.1.3', 'eventlet==0.24.1', 'flask-compress==1.4.0',
-        'flask-cors==3.0.8', 'pyasn1-modules==0.2.2', 'six==1.11.0', 'pymongo==3.10.1', 'bson==0.5.10', 'yapf==0.30.0'
+        'flask==1.0.2',
+        'flask-socketio==3.0.2',
+        'oauth2client==4.1.3',
+        'eventlet==0.24.1',
+        'flask-compress==1.4.0',
+        'flask-cors==3.0.8',
+        'pyasn1-modules==0.2.2',
+        'six==1.11.0',
+        'pymongo==3.10.1',
+        'bson==0.5.10',
+        'yapf==0.30.0',
+        'pytest==5.4.3',
+        'pytest-mock==3.1.1',
+        'pytest-env==0.6.2',
     ],
+    # yapf: enable
 )