Unverified Commit 2d7d0c97 authored by Ben Siraphob's avatar Ben Siraphob Committed by GitHub
Browse files

python3Packages.pyssim: modernize (#395720)

parents 010ec200 42e24c68

pkgs/development/python-modules/pyssim/default.nix

+11 −3
Original line number Diff line number Diff line
@@ -7,12 +7,17 @@ 
  pillow,

  pywavelets,

  fetchpatch,

  setuptools,

}:



buildPythonPackage rec {

  pname = "pyssim";

  version = "0.7";

  format = "setuptools";

  pyproject = true;



  build-system = [

    setuptools

  ];



  dependencies = [

    numpy
@@ -24,8 +29,8 @@ buildPythonPackage rec { 
  # PyPI tarball doesn't contain test images so let's use GitHub

  src = fetchFromGitHub {

    owner = "jterrace";

    repo = pname;

    rev = "v${version}";

    repo = "pyssim";

    tag = "v${version}";

    sha256 = "sha256-LDNIugQeRqNsAZ5ZxS/NxHokEAwefpfRutTRpR0IcXk=";

  };
@@ -39,17 +44,20 @@ buildPythonPackage rec { 


  # Tests are copied from .github/workflows/python-package.yml

  checkPhase = ''

    runHook preCheck

    $out/bin/pyssim test-images/test1-1.png test-images/test1-1.png | grep 1

    $out/bin/pyssim test-images/test1-1.png test-images/test1-2.png | grep 0.998

    $out/bin/pyssim test-images/test1-1.png "test-images/*" | grep -E " 1| 0.998| 0.672| 0.648" | wc -l | grep 4

    $out/bin/pyssim --cw --width 128 --height 128 test-images/test1-1.png test-images/test1-1.png | grep 1

    $out/bin/pyssim --cw --width 128 --height 128 test-images/test3-orig.jpg test-images/test3-rot.jpg | grep 0.938

    runHook postCheck

  '';



  meta = {

    description = "Module for computing Structured Similarity Image Metric (SSIM) in Python";

    mainProgram = "pyssim";

    homepage = "https://github.com/jterrace/pyssim";

    changelog = "https://github.com/jterrace/pyssim/blob/${src.tag}/CHANGES.md";

    license = lib.licenses.mit;

    maintainers = with lib.maintainers; [ jluttine ];

  };