add github hooks

.gitignore

@@ -6,3 +6,4 @@
 __pycache__/
 artifacts/
 benchmark/
+.pre-commit-config.yaml

.pre-commit-config.yaml

@@ -0,0 +1 @@
+/nix/store/9fa9jgr29n0zfbap74ac8hkcil8fgw8z-pre-commit-config.json

flake.nix

@@ -4,6 +4,7 @@
     systems.url = "github:nix-systems/x86_64-linux";
     nix-filter.url = "github:numtide/nix-filter";
     devshell.url = "github:numtide/devshell";
+    pre-commit-hooks.url = "github:cachix/pre-commit-hooks.nix";
     nix-appimage = {
       url = "github:ralismark/nix-appimage";
     };
@@ -12,9 +13,8 @@
     self,
     nixpkgs,
     systems,
-    nix-filter,
-    nix-appimage,
     devshell,
+    ...
   } @ inputs: let
     eachSystem = nixpkgs.lib.genAttrs (import systems);
     pkgsFor = eachSystem (system: ((nixpkgs.legacyPackages.${system}.extend devshell.overlays.default).extend self.overlays.default));
@@ -31,6 +31,12 @@
         lib = pkgs.lib;
       in
         {
+          pre-commit-check = inputs.pre-commit-hooks.lib.${system}.run {
+            src = ./.;
+            hooks = {
+              alejandra.enable = true;
+            };
+          };
           rs-tensor-encoder-test-runner-check =
             pkgs.runCommandNoCC "rs-encoder-test-runner-check" {
               nativeBuildInputs = [self.packages.${system}.rs-tensor-encoder-test-runner];
@@ -99,9 +105,11 @@
       buildPackage = inputs.self.packages.${system}.default;
     in {
       default = pkgs.mkShell {
+        inherit (self.checks.${system}.pre-commit-check) shellHook;
         nativeBuildInputs = buildPackage.nativeBuildInputs;
         buildInputs =
           buildPackage.buildInputs
+          ++ self.checks.${system}.pre-commit-check.enabledPackages
           ++ (with pkgs; [
             # glibc
             rustc

integration_tests/test_python_integration.py

@@ -79,7 +79,7 @@ def read_all_raster_bands(path):
     as a string and the value the array data
     """
     with rasterio.open(path) as r:
-        return {str(i): r.read(i) for i in range(1, r.count + 1)}
+        return {f"B{i}": r.read(i) for i in range(1, r.count + 1)}
 
 def test_python_hyspecnet_integration(hyspecnet_root, encoded_hyspecnet_path):
     source_file_data = {file: read_all_raster_bands(file) for file in hyspecnet_root.glob("**/*SPECTRAL_IMAGE.TIF")}