black

panama/read.py

@@ -307,15 +307,15 @@ def read_DAT(
         )
 
         pdgids = df_particles["pdgid"].unique()
-        
+
         mass_map = dict()
         for pdgid in pdgids:
             if pdgid == pdg_error_val:
                 mass_map[pdgid] = 0
             else:
                 mass = Particle.from_pdgid(pdgid).mass
-                mass_map[pdgid] = mass / 1000 if mass is not None else 0 # GeV
-        
+                mass_map[pdgid] = mass / 1000 if mass is not None else 0  # GeV
+
         df_particles["mass"] = df_particles["pdgid"].map(mass_map, na_action=None)
         df_particles["energy"] = df_particles.eval("sqrt(mass**2+px**2+py**2+pz**2)")
         df_particles["zenith"] = df_particles.eval("arccos(pz/sqrt(px**2+py**2+pz**2))")

pyproject.toml

@@ -24,6 +24,7 @@ crflux = "^1.0.6"
 black = "^22.10.0"
 pytest = "^7.2.0"
 coverage = "^7.1.0"
+pre-commit = "^3.0.4"
 
 [tool.poetry.scripts]
 panama = 'panama.cli:cli'

tests/test_run.py

@@ -5,6 +5,7 @@
 from click.testing import CliRunner
 from panama import read_DAT
 
+
 def test_run_fail(
     test_file_path=Path(__file__).parent / "files" / "example_corsika.template",
 ):
@@ -46,34 +47,36 @@ def test_cli_missing_executable(
 
 def test_cli(
     test_file_path=Path(__file__).parent / "files" / "example_corsika.template",
-    corsika_path=Path(__file__).parent.parent / "corsika-77420" / "run" / "corsika77420Linux_SIBYLL_urqmd",
-    compare_files=Path(__file__).parent / "files" / "compare" / "DAT*"
+    corsika_path=Path(__file__).parent.parent
+    / "corsika-77420"
+    / "run"
+    / "corsika77420Linux_SIBYLL_urqmd",
+    compare_files=Path(__file__).parent / "files" / "compare" / "DAT*",
 ):
     runner = CliRunner()
     result = runner.invoke(
         cli,
         [
             f"{test_file_path}",
             "--primary",
-            "{2212: 1, 1000260560: 1}", # proton and iron
+            "{2212: 1, 1000260560: 1}",  # proton and iron
             "--corsika",
             f"{corsika_path}",
             "--output",
             "corsika_output",
             "--seed",
             "137",
             "--jobs",
-            "1" # this also tests multi threading since we have one job per primary
-        ]
+            "1",  # this also tests multi threading since we have one job per primary
+        ],
     )
-    
+
     assert "cleanup now" in result.output
     assert result.exit_code == 0
-    
+
     run_header, event_header, ps = read_DAT(glob=compare_files)
     run_header_2, event_header_2, ps_2 = read_DAT(glob="corsika_output/DAT*")
 
     assert run_header.equals(run_header_2)
     assert event_header.equals(event_header_2)
     assert ps.equals(ps_2)
-