1 files changed, 9 insertions, 0 deletions
diff --git a/nixpkgs/pkgs/development/python-modules/mdformat-beautysh/default.nix b/nixpkgs/pkgs/development/python-modules/mdformat-beautysh/default.nix
index 522494f783fe..a117fea936eb 100644
--- a/nixpkgs/pkgs/development/python-modules/mdformat-beautysh/default.nix
+++ b/nixpkgs/pkgs/development/python-modules/mdformat-beautysh/default.nix
@@ -6,6 +6,7 @@
 , mdformat-gfm
 , mdit-py-plugins
 , poetry-core
+, pytestCheckHook
 , pythonOlder
 }:
 
@@ -37,6 +38,14 @@ buildPythonPackage rec {
     beautysh
   ];
 
+  nativeCheckInputs = [
+    pytestCheckHook
+  ];
+
+  pythonImportsCheck = [
+    "mdformat_beautysh"
+  ];
+
   meta = with lib; {
     description = "Mdformat plugin to beautify Bash scripts";
     homepage = "https://github.com/hukkin/mdformat-beautysh";