diff --git a/pkgs/development/python-modules/pygtrie/default.nix b/pkgs/development/python-modules/pygtrie/default.nix
index 8e7d4d823b81..b8fcfb686e90 100644
--- a/pkgs/development/python-modules/pygtrie/default.nix
+++ b/pkgs/development/python-modules/pygtrie/default.nix
@@ -3,6 +3,7 @@
   fetchPypi,
   buildPythonPackage,
   setuptools,
+  pytestCheckHook,
 }:
 buildPythonPackage (finalAttrs: {
   pname = "pygtrie";
@@ -19,6 +20,10 @@ buildPythonPackage (finalAttrs: {
 
   build-system = [ setuptools ];
 
+  nativeCheckInputs = [ pytestCheckHook ];
+
+  enabledTestPaths = [ "test.py" ];
+
   meta = {
     homepage = "https://github.com/mina86/pygtrie";
     description = "Trie data structure implementation";