pythonPackages.packet-python: enable tests

1 files changed, 8 insertions, 4 deletions

diff --git a/pkgs/development/python-modules/packet-python/default.nix b/pkgs/development/python-modules/packet-python/default.nix
index e233dfeb086a..61d387590480 100644
--- a/pkgs/development/python-modules/packet-python/default.nix
+++ b/pkgs/development/python-modules/packet-python/default.nix
@@ -5,7 +5,9 @@
 , python
 
 # For tests/setup.py
+, pytest
 , pytestrunner
+, requests-mock
 }:
 
 buildPythonPackage rec {
@@ -17,14 +19,16 @@ buildPythonPackage rec {
   };
   nativeBuildInputs = [ pytestrunner ];
   propagatedBuildInputs = [ requests ];
+  checkInputs = [
+    pytest
+    pytestrunner
+    requests-mock
+  ];
 
   checkPhase = ''
-    ${python.interpreter} -m unittest discover -s test
+    ${python.interpreter} setup.py test
   '';
 
-  # Not all test files are included in archive
-  doCheck = false;
-
   meta = {
     description = "A Python client for the Packet API.";
     homepage    = "https://github.com/packethost/packet-python";