nixpkgs/nixos/tests/nixos-test-driver/extra-python-packages.nix

{ ... }:
{
  name = "extra-python-packages";

  extraPythonPackages = p: [ p.numpy ];

  nodes = { };

  testScript = ''
    import numpy as np
    assert str(np.zeros(4) == "array([0., 0., 0., 0.])")
  '';
}