Skip to content

Commit 70feb35

Browse files
committed
Update tests
1 parent 8a460ba commit 70feb35

File tree

1 file changed

+1
-24
lines changed

1 file changed

+1
-24
lines changed

tests/unit/schema/v1/test_template_builder.py

Lines changed: 1 addition & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -234,22 +234,6 @@ def test_coordinate_with_chunk_grid() -> None:
234234
assert cdp_var.metadata.chunk_grid.name == "regular"
235235
assert cdp_var.metadata.chunk_grid.configuration.chunk_shape == [20, 20]
236236

237-
def test_coordinate_with_stats() -> None:
238-
"""Test adding coordinates with stats."""
239-
builder = MDIODatasetBuilder("test_dataset")
240-
builder.add_dimension("inline", 100)
241-
builder.add_dimension("crossline", 100)
242-
243-
# Add coordinate with stats
244-
builder.add_coordinate("cdp", dimensions=["inline", "crossline"], metadata={"statsV1": {"count": 100, "sum": 1215.1, "sumSquares": 125.12, "min": 5.61, "max": 10.84, "histogram": {"binCenters": [1, 2], "counts": [10, 15]}}})
245-
246-
assert len(builder._variables) == 2
247-
assert len(builder._coordinates) == 1
248-
cdp_var = builder._coordinates[0]
249-
assert cdp_var.name == "cdp"
250-
assert cdp_var.data_type == ScalarType.FLOAT32
251-
assert cdp_var.metadata.stats_v1.count == 100
252-
assert cdp_var.metadata.stats_v1.sum == 1215.1
253237

254238
def test_coordinate_with_full_metadata() -> None:
255239
"""Test adding coordinates with all metadata."""
@@ -258,7 +242,7 @@ def test_coordinate_with_full_metadata() -> None:
258242
builder.add_dimension("crossline", 100)
259243

260244
# Add coordinate with all metadata
261-
builder.add_coordinate("cdp", dimensions=["inline", "crossline"], metadata={"unitsV1": {"length": "m"}, "attributes": {"MGA": 51}, "chunkGrid": {"name": "regular", "configuration": {"chunkShape": [20]}}, "statsV1": {"count": 100, "sum": 1215.1, "sumSquares": 125.12, "min": 5.61, "max": 10.84, "histogram": {"binCenters": [1, 2], "counts": [10, 15]}}})
245+
builder.add_coordinate("cdp", dimensions=["inline", "crossline"], metadata={"unitsV1": {"length": "m"}, "attributes": {"MGA": 51}, "chunkGrid": {"name": "regular", "configuration": {"chunkShape": [20]}}})
262246

263247
assert len(builder._variables) == 2
264248
assert len(builder._coordinates) == 1
@@ -269,13 +253,6 @@ def test_coordinate_with_full_metadata() -> None:
269253
assert cdp_var.metadata.attributes["MGA"] == 51
270254
assert cdp_var.metadata.chunk_grid.name == "regular"
271255
assert cdp_var.metadata.chunk_grid.configuration.chunk_shape == [20]
272-
assert cdp_var.metadata.stats_v1.count == 100
273-
assert cdp_var.metadata.stats_v1.sum == 1215.1
274-
assert cdp_var.metadata.stats_v1.sum_squares == 125.12
275-
assert cdp_var.metadata.stats_v1.min == 5.61
276-
assert cdp_var.metadata.stats_v1.max == 10.84
277-
assert cdp_var.metadata.stats_v1.histogram.bin_centers == [1, 2]
278-
assert cdp_var.metadata.stats_v1.histogram.counts == [10, 15]
279256

280257
j = builder.build().json()
281258
print(j)

0 commit comments

Comments
 (0)