commit 2d7d0c9733194d9a30d8949d9dc265ed992e72a7 · pyrox.dev/nixpkgs

+11 -3
pkgs/development/python-modules/pyssim/default.nix
···

       7
       7
        
         pillow,

     

       8
       8
        
         pywavelets,

     

       9
       9
        
         fetchpatch,

     

       10
       10
       +
         setuptools,

     

       10
       11
        
       }:

     

       11
       12
        
       

     

       12
       13
        
       buildPythonPackage rec {

     

       13
       14
        
         pname = "pyssim";

     

       14
       15
        
         version = "0.7";

     

       15
       15
       -
         format = "setuptools";

     

       16
       16
       +
         pyproject = true;

     

       17
       17
       +
       

     

       18
       18
       +
         build-system = [

     

       19
       19
       +
           setuptools

     

       20
       20
       +
         ];

     

       16
       21
        
       

     

       17
       22
        
         dependencies = [

     

       18
       23
        
           numpy

     
···

       24
       29
        
         # PyPI tarball doesn't contain test images so let's use GitHub

     

       25
       30
        
         src = fetchFromGitHub {

     

       26
       31
        
           owner = "jterrace";

     

       27
       27
       -
           repo = pname;

     

       28
       28
       -
           rev = "v${version}";

     

       32
       32
       +
           repo = "pyssim";

     

       33
       33
       +
           tag = "v${version}";

     

       29
       34
        
           sha256 = "sha256-LDNIugQeRqNsAZ5ZxS/NxHokEAwefpfRutTRpR0IcXk=";

     

       30
       35
        
         };

     

       31
       36
        
       

     
···

       39
       44
        
       

     

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

     

       41
       46
        
         checkPhase = ''

     

       47
       47
       +
           runHook preCheck

     

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

     

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

     

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

     

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

     

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

     

       53
       53
       +
           runHook postCheck

     

       47
       54
        
         '';

     

       48
       55
        
       

     

       49
       56
        
         meta = {

     

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

     

       51
       58
        
           mainProgram = "pyssim";

     

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

     

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

     

       53
       61
        
           license = lib.licenses.mit;

     

       54
       62
        
           maintainers = with lib.maintainers; [ jluttine ];

     

       55
       63
        
         };