feat: polish CSV header format with improved metadata and column naming

This commit is contained in:
2026-07-03 14:58:19 +03:00
parent 876e1e3ab8
commit a7eef7bc20
7 changed files with 86 additions and 577 deletions
+4 -4
View File
@@ -464,11 +464,11 @@ class TestMultiBranchCSVExport:
df = pd.read_csv(csv_path, comment="#")
# Should have total spectrum column
assert "spectrum" in df.columns
assert "total_spectrum" in df.columns
# Should have per-branch spectrum columns (intensity/endpoint/transition are in the header)
assert "branch_1_spectrum" in df.columns
assert "branch_2_spectrum" in df.columns
assert "branch 1" in df.columns
assert "branch 2" in df.columns
# Should NOT have repeated per-branch metadata columns
assert "branch_1_intensity" not in df.columns
@@ -502,7 +502,7 @@ class TestMultiBranchCSVExport:
df = pd.read_csv(csv_path, comment="#")
assert "spectrum" in df.columns
assert "total_spectrum" in df.columns
assert "Fermi" in df.columns
+10 -4
View File
@@ -154,7 +154,7 @@ class TestCSVMetadataHeader:
lines = f.readlines()
assert any("# beta-spectrum v" in line for line in lines)
assert any("0.3.0" in line for line in lines)
assert any("0.4.0" in line for line in lines)
os.unlink(filename)
def test_header_contains_timestamp(self):
@@ -236,7 +236,9 @@ class TestCSVMetadataHeader:
with open(filename, "r") as f:
content = f.read()
assert "# git_commit:" in content
# Check for version with build hash instead of git_commit line
assert "# beta-spectrum v" in content
assert "(build." in content
os.unlink(filename)
def test_header_contains_corrections(self):
@@ -264,8 +266,12 @@ class TestCSVMetadataHeader:
content = f.read()
assert "# corrections:" in content
assert "phase_space" in content
assert "fermi" in content
# phase_space and fermi are basic components and should not be listed
assert "screening" in content
assert "finite_size" in content
assert "charge_dist" in content
assert "radiative" in content
assert "exchange" in content
os.unlink(filename)