Skip to content

Commit 1df1325

Browse files
committed
test: enhance grid export tests to verify header and data lines for missing scalar data
1 parent 15f2042 commit 1df1325

1 file changed

Lines changed: 21 additions & 4 deletions

File tree

tests/qgis/test_grid_export.py

Lines changed: 21 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -121,19 +121,36 @@ def test_export_grid_ascii_no_scalars(self):
121121
with open(temp_path, 'r') as f:
122122
lines = f.readlines()
123123

124-
# Should still create file with zeros
124+
# Should create file with header
125+
self.assertGreater(len(lines), 0)
125126
self.assertIn("# ASCII Grid Export: test_grid_no_scalars", lines[0])
126-
self.assertGreaterEqual(len(lines), 9) # Header + 4 data lines
127127

128-
# Verify that values are zero
129-
data_lines = lines[5:]
128+
# Count non-header lines (data lines)
129+
header_count = 0
130+
for line in lines:
131+
if line.strip().startswith("#"):
132+
header_count += 1
133+
else:
134+
break
135+
136+
# We should have at least some header and data lines
137+
self.assertGreater(header_count, 0)
138+
139+
# Verify data lines have zeros for missing scalar data
140+
data_lines = lines[header_count:]
141+
data_count = 0
130142
for line in data_lines:
131143
if line.strip():
132144
parts = line.strip().split()
133145
if len(parts) == 4:
146+
data_count += 1
134147
value = float(parts[3])
148+
# Should be zero for missing scalar data
135149
self.assertAlmostEqual(value, 0.0, places=5)
136150

151+
# Should have data for the 4 cells
152+
self.assertEqual(data_count, 4, f"Expected 4 data lines, got {data_count}")
153+
137154
finally:
138155
Path(temp_path).unlink(missing_ok=True)
139156

0 commit comments

Comments
 (0)