feat: polish CSV header format with improved metadata and column naming
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
|
||||
@@ -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)
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user