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

{
  lib,
  blas,
  lapack,
  buildPythonPackage,
  cffi,
  fetchFromGitHub,
  nose,
  numpy,
  stdenv,
}:

buildPythonPackage {
  pname = "prox-tv";
  version = "3.3.0";
  format = "setuptools";

  src = fetchFromGitHub {
    owner = "albarji";
    repo = "proxTV";
    rev = "e621585d5aaa7983fbee68583f7deae995d3bafb";
    sha256 = "0mlrjbb5rw78dgijkr3bspmsskk6jqs9y7xpsgs35i46dvb327q5";
  };

  nativeCheckInputs = [ nose ];

  propagatedBuildInputs = [
    numpy
    cffi
  ];

  # this test is known to fail on darwin
  checkPhase = ''
    nosetests --exclude=test_tvp_1d ${lib.optionalString stdenv.isDarwin " --exclude=test_tv2_1d"}
  '';

  propagatedNativeBuildInputs = [ cffi ];

  buildInputs = [
    blas
    lapack
  ];

  enableParallelBuilding = true;

  meta = with lib; {
    homepage = "https://github.com/albarji/proxTV";
    description = "Toolbox for fast Total Variation proximity operators";
    license = licenses.bsd2;
    maintainers = with maintainers; [ multun ];
  };
}