nixpkgs/pkgs/development/python-modules/h5py/default.nix
2021-04-09 02:13:46 +02:00

56 lines
1.7 KiB
Nix

{ lib, fetchPypi, isPy27, python, buildPythonPackage, pythonOlder
, numpy, hdf5, cython, six, pkgconfig, unittest2, fetchpatch
, mpi4py ? null, openssh, pytestCheckHook, cached-property }:
assert hdf5.mpiSupport -> mpi4py != null && hdf5.mpi == mpi4py.mpi;
let
mpi = hdf5.mpi;
mpiSupport = hdf5.mpiSupport;
in buildPythonPackage rec {
version = "3.2.1";
pname = "h5py";
disabled = isPy27;
src = fetchPypi {
inherit pname version;
sha256 = "sha256-iUdL6RG/zbNMvw2YuOxItXjCeon9sa5O51E/HvjZJJ4=";
};
# avoid strict pinning of numpy
postPatch = ''
substituteInPlace setup.py \
--replace "numpy ==" "numpy >="
'';
HDF5_DIR = "${hdf5}";
HDF5_MPI = if mpiSupport then "ON" else "OFF";
postConfigure = ''
# Needed to run the tests reliably. See:
# https://bitbucket.org/mpi4py/mpi4py/issues/87/multiple-test-errors-with-openmpi-30
${lib.optionalString mpiSupport "export OMPI_MCA_rmaps_base_oversubscribe=yes"}
'';
preBuild = if mpiSupport then "export CC=${mpi}/bin/mpicc" else "";
# tests now require pytest-mpi, which isn't available and difficult to package
doCheck = false;
checkInputs = lib.optional isPy27 unittest2 ++ [ pytestCheckHook openssh ];
nativeBuildInputs = [ pkgconfig cython ];
buildInputs = [ hdf5 ]
++ lib.optional mpiSupport mpi;
propagatedBuildInputs = [ numpy six]
++ lib.optionals mpiSupport [ mpi4py openssh ]
++ lib.optionals (pythonOlder "3.8") [ cached-property ];
pythonImportsCheck = [ "h5py" ];
meta = with lib; {
description = "Pythonic interface to the HDF5 binary data format";
homepage = "http://www.h5py.org/";
license = licenses.bsd3;
maintainers = [ ];
};
}