depot/third_party/nixpkgs/pkgs/development/python-modules/testbook/default.nix

{
  lib,
  buildPythonPackage,
  pythonOlder,
  fetchFromGitHub,
  nbformat,
  nbclient,
  ipykernel,
  pandas,
  pytestCheckHook,
  setuptools,
  traitlets,
}:

buildPythonPackage rec {
  pname = "testbook";
  version = "0.4.2";
  format = "pyproject";

  disabled = pythonOlder "3.6";

  src = fetchFromGitHub {
    owner = "nteract";
    repo = pname;
    rev = version;
    hash = "sha256-qaDgae/5TRpjmjOf7aom7TC5HLHp0PHM/ds47AKtq8U=";
  };

  propagatedBuildInputs = [
    nbclient
    nbformat
  ];

  nativeBuildInputs = [ setuptools ];

  nativeCheckInputs = [
    ipykernel
    pandas
    pytestCheckHook
    traitlets
  ];

  pythonImportsCheck = [ "testbook" ];

  meta = with lib; {
    description = "A unit testing framework extension for testing code in Jupyter Notebooks";
    homepage = "https://testbook.readthedocs.io/";
    license = with licenses; [ bsd3 ];
    maintainers = with maintainers; [ djacu ];
  };
}