format nix and setup pythonpath in shellHook

Author: nbren12

default.nix

@@ -1,28 +1,29 @@
-{
-    lib,
-    stdenv,
-    gfortran,
-    cmake,
-    pfunit,
-    pythonPackages,
-    llvmPackages
-} :
+{ lib, stdenv, gfortran, cmake, pfunit, pythonPackages, llvmPackages }:
 stdenv.mkDerivation {
-    name = "call_py_fort";
-    src = ./.;
+  name = "call_py_fort";
+  src = ./.;
 
-    nativeBuildInputs = [ pfunit ] ++ lib.optional stdenv.isDarwin llvmPackages.openmp;
-    buildInputs = [ gfortran gfortran.cc.lib cmake  gfortran.cc pfunit  pythonPackages.python ];
-    propagatedBuildInputs = [ pythonPackages.cffi pythonPackages.numpy ] ;
-    doCheck = true;
+  nativeBuildInputs = [ pfunit ]
+    ++ lib.optional stdenv.isDarwin llvmPackages.openmp;
+  buildInputs =
+    [ gfortran gfortran.cc.lib cmake gfortran.cc pfunit pythonPackages.python ];
+  propagatedBuildInputs = [ pythonPackages.cffi pythonPackages.numpy ];
+  doCheck = true;
 
-    preCheck = ''
-        export PYTHONPATH=$(pwd)/../test:$PYTHONPATH
-        # for mac
-        export DYLD_LIBRARY_PATH=$(pwd)/src
-        # for linux
-        export LD_LIBRARY_PATH=$(pwd)/src
-    '';
-    inherit pfunit;
+  preCheck = ''
+    export PYTHONPATH=$(pwd)/../test:$PYTHONPATH
+    # for mac
+    export DYLD_LIBRARY_PATH=$(pwd)/src
+    # for linux
+    export LD_LIBRARY_PATH=$(pwd)/src
+  '';
+  shellHook = ''
+    export PYTHONPATH=$(pwd)/test:$PYTHONPATH
+    # for mac
+    export DYLD_LIBRARY_PATH=$(pwd)/src
+    # for linux
+    export LD_LIBRARY_PATH=$(pwd)/src
+  '';
+  inherit pfunit;
 
 }

pfunit.nix

@@ -1,21 +1,14 @@
-{
-    stdenv,
-    m4,
-    fetchFromGitHub,
-    gfortran,
-    cmake,
-    python3
-} :
+{ stdenv, m4, fetchFromGitHub, gfortran, cmake, python3 }:
 stdenv.mkDerivation {
-    name = "pFUnit";
-    src = fetchFromGitHub {
-        owner = "Goddard-Fortran-Ecosystem";
-        repo = "pFUnit";
-        rev = "v4.2.1";
-        fetchSubmodules = true;
-        sha256 = "sha256-yjuJHvJ83PAQBDDk7TD4b6VWGoKbiexdjBCTLsNIIxI=";
+  name = "pFUnit";
+  src = fetchFromGitHub {
+    owner = "Goddard-Fortran-Ecosystem";
+    repo = "pFUnit";
+    rev = "v4.2.1";
+    fetchSubmodules = true;
+    sha256 = "sha256-yjuJHvJ83PAQBDDk7TD4b6VWGoKbiexdjBCTLsNIIxI=";
 
-    };
-    buildInputs = [ python3 gfortran cmake gfortran.cc m4 ];
+  };
+  buildInputs = [ python3 gfortran cmake gfortran.cc m4 ];
 
 }

shell.nix

@@ -1,5 +1,7 @@
-let 
-    nixpkgs = import <nixpkgs> { };
-    pfunit = nixpkgs.callPackage ./pfunit.nix { };
-in
-    nixpkgs.callPackage ./default.nix { inherit pfunit; pythonPackages = nixpkgs.python3Packages; }
+let
+  nixpkgs = import <nixpkgs> { };
+  pfunit = nixpkgs.callPackage ./pfunit.nix { };
+in nixpkgs.callPackage ./default.nix {
+  inherit pfunit;
+  pythonPackages = nixpkgs.python3Packages;
+}