Skip to content

Commit 2da102d

Browse files
committed
Fix leftover debugging code and fix linter errors
1 parent bbc88e4 commit 2da102d

File tree

1 file changed

+3
-17
lines changed

1 file changed

+3
-17
lines changed

tests/unit/benchmark/test_output.py

Lines changed: 3 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -34,20 +34,10 @@ def test_generative_benchmark_marshalling():
3434
report = GenerativeBenchmarksReport(benchmarks=[mock_benchmark])
3535

3636
serialized = report.model_dump()
37-
print("Serialized: ", serialized)
3837
deserialized = GenerativeBenchmarksReport.model_validate(serialized)
39-
print("Deserialized: ", deserialized)
4038
deserialized_benchmark = deserialized.benchmarks[0]
41-
serialized = deserialized.model_dump()
42-
print("Re-serialized: ", serialized)
4339

44-
print("Fields:", {name: getattr(deserialized, name) for name in deserialized.model_fields})
45-
print("Extras:", deserialized.__pydantic_extra__)
46-
47-
48-
#for field in mock_benchmark.model_fields_set:
49-
# assert getattr(mock_benchmark, field) == getattr(deserialized_benchmark, field)
50-
# Workaround for duplicate fields for computed fields.
40+
# model_dump as workaround for duplicate fields for computed fields.
5141
assert mock_benchmark.model_dump() == deserialized_benchmark.model_dump()
5242

5343

@@ -65,9 +55,7 @@ def test_file_json():
6555
loaded_report = GenerativeBenchmarksReport.load_file(mock_path)
6656
loaded_benchmark = loaded_report.benchmarks[0]
6757

68-
#for field in mock_benchmark.model_fields_set:
69-
# assert getattr(mock_benchmark, field) == getattr(loaded_benchmark, field)
70-
# Workaround for duplicate fields for computed fields.
58+
# model_dump as workaround for duplicate fields for computed fields.
7159
assert mock_benchmark.model_dump() == loaded_benchmark.model_dump()
7260

7361
mock_path.unlink()
@@ -87,9 +75,7 @@ def test_file_yaml():
8775
loaded_report = GenerativeBenchmarksReport.load_file(mock_path)
8876
loaded_benchmark = loaded_report.benchmarks[0]
8977

90-
#for field in mock_benchmark.model_fields_set:
91-
# assert getattr(mock_benchmark, field) == getattr(loaded_benchmark, field)
92-
# Workaround for duplicate fields for computed fields.
78+
# model_dump as workaround for duplicate fields for computed fields.
9379
assert mock_benchmark.model_dump() == loaded_benchmark.model_dump()
9480

9581
mock_path.unlink()

0 commit comments

Comments
 (0)