depot/third_party/nixpkgs/pkgs/development/python-modules/gumath/default.nix
Default email 7e47f3658e Project import generated by Copybara.
GitOrigin-RevId: 1925c603f17fc89f4c8f6bf6f631a802ad85d784
2024-09-26 11:04:55 +00:00

66 lines
1.9 KiB
Nix

{
lib,
stdenv,
buildPythonPackage,
fetchpatch,
python,
numba,
ndtypes,
xnd,
libndtypes,
libxnd,
libgumath,
isPy27,
}:
buildPythonPackage {
pname = "gumath";
format = "setuptools";
disabled = isPy27;
inherit (libgumath) src version meta;
patches = [
# https://github.com/xnd-project/gumath/pull/42
(fetchpatch {
name = "remove-np-warnings-call.patch";
url = "https://github.com/xnd-project/gumath/commit/83ab3aa3b07d55654b4e6e75e5ec6be8190fca97.patch";
hash = "sha256-7lUXNVH5M+Go1iEu0bud03XI8cyGbdLNdLraMZplDaM=";
})
(fetchpatch {
name = "remove-np-1.25-bartlett-test-assertion.patch";
url = "https://github.com/xnd-project/gumath/commit/8741e31f2967ded08c96a7f0631e1e38fe813870.patch";
hash = "sha256-flltk3RNPHalbcIV0BrkxWuhqqJBrycos7Fyv3P3mWg=";
})
];
nativeCheckInputs = [ numba ];
propagatedBuildInputs = [
ndtypes
xnd
];
postPatch = ''
substituteInPlace setup.py \
--replace 'add_include_dirs = [".", "libgumath", "ndtypes/python/ndtypes", "xnd/python/xnd"] + INCLUDES' \
'add_include_dirs = [".", "${libndtypes.dev}/include", "${libxnd}/include", "${libgumath}/include"]' \
--replace 'add_library_dirs = ["libgumath", "ndtypes/libndtypes", "xnd/libxnd"] + LIBS' \
'add_library_dirs = ["${libndtypes}/lib", "${libxnd}/lib", "${libgumath}/lib"]' \
--replace 'add_runtime_library_dirs = ["$ORIGIN"]' \
'add_runtime_library_dirs = ["${libndtypes}/lib", "${libxnd}/lib", "${libgumath}/lib"]'
'';
postInstall = lib.optionalString stdenv.hostPlatform.isDarwin ''
install_name_tool -add_rpath ${libgumath}/lib $out/${python.sitePackages}/gumath/_gumath.*.so
'';
checkPhase = ''
pushd python
mv gumath _gumath
# minor precision issues
substituteInPlace test_gumath.py --replace 'test_sin' 'dont_test_sin'
python test_gumath.py
python test_xndarray.py
popd
'';
}