44
55import pytest
66
7- from agon import AGON
8-
97
108@pytest .fixture
119def simple_data () -> list [dict [str , Any ]]:
@@ -17,12 +15,6 @@ def simple_data() -> list[dict[str, Any]]:
1715 ]
1816
1917
20- @pytest .fixture
21- def simple_config (simple_data : list [dict [str , Any ]]) -> dict [str , Any ]:
22- """Trained config for simple_data."""
23- return AGON .train (simple_data , "test" )
24-
25-
2618@pytest .fixture
2719def nested_data () -> list [dict [str , Any ]]:
2820 """Test data with nested objects."""
@@ -32,12 +24,6 @@ def nested_data() -> list[dict[str, Any]]:
3224 ]
3325
3426
35- @pytest .fixture
36- def nested_config (nested_data : list [dict [str , Any ]]) -> dict [str , Any ]:
37- """Trained config for nested_data."""
38- return AGON .train (nested_data , "nested" )
39-
40-
4127@pytest .fixture
4228def list_data () -> list [dict [str , Any ]]:
4329 """Test data with nested lists."""
@@ -47,12 +33,6 @@ def list_data() -> list[dict[str, Any]]:
4733 ]
4834
4935
50- @pytest .fixture
51- def list_config (list_data : list [dict [str , Any ]]) -> dict [str , Any ]:
52- """Trained config for list_data."""
53- return AGON .train (list_data , "list_test" )
54-
55-
5636@pytest .fixture
5737def data_with_nulls () -> list [dict [str , Any ]]:
5838 """Test data with explicit nulls and missing fields."""
@@ -61,9 +41,3 @@ def data_with_nulls() -> list[dict[str, Any]]:
6141 {"id" : 2 , "name" : "Bob" }, # Missing role
6242 {"id" : 3 }, # Missing name and role
6343 ]
64-
65-
66- @pytest .fixture
67- def nulls_config (data_with_nulls : list [dict [str , Any ]]) -> dict [str , Any ]:
68- """Trained config for data_with_nulls."""
69- return AGON .train (data_with_nulls , "nulls" )
0 commit comments