Skip to content

Commit 4a02328

Browse files
IndexSeekIndexSeek
committed
test: deduplicate pytest params for problem 570
1 parent 9f6a494 commit 4a02328

File tree

3 files changed

+45
-126
lines changed

3 files changed

+45
-126
lines changed

‎tests/test_pandas.py

Lines changed: 1 addition & 41 deletions
Original file line numberDiff line numberDiff line change
@@ -86,47 +86,7 @@ def test_problem_550(input_data, expected_data):
8686

8787
@pytest.mark.parametrize(
8888
"input_data, expected_data",
89-
[
90-
pytest.param(
91-
{
92-
"id": [101, 102, 103, 104, 105, 106],
93-
"name": ["John", "Dan", "James", "Amy", "Anne", "Ron"],
94-
"managerId": [None, 101, 101, 101, 101, 101],
95-
},
96-
{"name": ["John"]},
97-
id="happy_path_one_with_five",
98-
),
99-
pytest.param(
100-
{
101-
"id": [101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111],
102-
"name": [
103-
"John",
104-
"Dan",
105-
"James",
106-
"Amy",
107-
"Anne",
108-
"Ron",
109-
"Alex",
110-
"Bob",
111-
"Charles",
112-
"David",
113-
"Edward",
114-
],
115-
"managerId": [None, 101, 101, 101, 101, 101, 110, 110, 110, 110, 110],
116-
},
117-
{"name": ["John", "David"]},
118-
id="happy_path_two_with_five",
119-
),
120-
pytest.param(
121-
{
122-
"id": [101, 102, 103, 104, 105, 106],
123-
"name": ["John", "Dan", "James", "Amy", "Anne", "Ron"],
124-
"managerId": [None, 101, 101, 101, 101, 102],
125-
},
126-
{"name": []},
127-
id="none_with_five",
128-
),
129-
],
89+
PARAMS_PROBLEM_570,
13090
)
13191
def test_problem_570(input_data, expected_data):
13292
table = pd.DataFrame(input_data)

‎tests/test_problem_params.py

Lines changed: 42 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -243,3 +243,45 @@
243243
id="edge_case_single_entry",
244244
),
245245
]
246+
247+
PARAMS_PROBLEM_570 = [
248+
pytest.param(
249+
{
250+
"id": [101, 102, 103, 104, 105, 106],
251+
"name": ["John", "Dan", "James", "Amy", "Anne", "Ron"],
252+
"managerId": [None, 101, 101, 101, 101, 101],
253+
},
254+
{"name": ["John"]},
255+
id="happy_path_one_with_five",
256+
),
257+
pytest.param(
258+
{
259+
"id": [101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111],
260+
"name": [
261+
"John",
262+
"Dan",
263+
"James",
264+
"Amy",
265+
"Anne",
266+
"Ron",
267+
"Alex",
268+
"Bob",
269+
"Charles",
270+
"David",
271+
"Edward",
272+
],
273+
"managerId": [None, 101, 101, 101, 101, 101, 110, 110, 110, 110, 110],
274+
},
275+
{"name": ["John", "David"]},
276+
id="happy_path_two_with_five",
277+
),
278+
pytest.param(
279+
{
280+
"id": [101, 102, 103, 104, 105, 106],
281+
"name": ["John", "Dan", "James", "Amy", "Anne", "Ron"],
282+
"managerId": [None, 101, 101, 101, 101, 102],
283+
},
284+
{"name": []},
285+
id="none_with_five",
286+
),
287+
]

‎tests/test_pyarrow.py

Lines changed: 2 additions & 85 deletions
Original file line numberDiff line numberDiff line change
@@ -71,50 +71,7 @@ def test_problem_197(input_data, expected_data):
7171

7272
@pytest.mark.parametrize(
7373
"input_data, expected_data",
74-
[
75-
pytest.param(
76-
{
77-
"player_id": [1, 1, 2, 3, 3],
78-
"device_id": [2, 2, 3, 1, 4],
79-
"event_date": [
80-
datetime(2016, 3, 1),
81-
datetime(2016, 3, 2),
82-
datetime(2017, 6, 25),
83-
datetime(2016, 3, 2),
84-
datetime(2018, 7, 3),
85-
],
86-
"games_played": [5, 6, 1, 0, 5],
87-
},
88-
{"fraction": [0.33]},
89-
id="happy_path_basic",
90-
),
91-
pytest.param(
92-
{
93-
"player_id": [1, 1, 1, 2, 2],
94-
"device_id": [2, 2, 3, 1, 4],
95-
"event_date": [
96-
datetime(2023, 1, 1),
97-
datetime(2023, 1, 2),
98-
datetime(2023, 1, 3),
99-
datetime(2023, 1, 1),
100-
datetime(2023, 1, 2),
101-
],
102-
"games_played": [1, 2, 3, 4, 5],
103-
},
104-
{"fraction": [1.0]},
105-
id="happy_path_multiple_dates",
106-
),
107-
pytest.param(
108-
{
109-
"player_id": [1],
110-
"device_id": [1],
111-
"event_date": [datetime(2023, 1, 1)],
112-
"games_played": [1],
113-
},
114-
{"fraction": [0.0]},
115-
id="edge_case_single_entry",
116-
),
117-
],
74+
PARAMS_PROBLEM_550,
11875
)
11976
def test_problem_550(input_data, expected_data):
12077
table = pa.Table.from_pydict(input_data)
@@ -125,47 +82,7 @@ def test_problem_550(input_data, expected_data):
12582

12683
@pytest.mark.parametrize(
12784
"input_data, expected_data",
128-
[
129-
pytest.param(
130-
{
131-
"id": [101, 102, 103, 104, 105, 106],
132-
"name": ["John", "Dan", "James", "Amy", "Anne", "Ron"],
133-
"managerId": [None, 101, 101, 101, 101, 101],
134-
},
135-
{"name": ["John"]},
136-
id="happy_path_one_with_five",
137-
),
138-
pytest.param(
139-
{
140-
"id": [101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111],
141-
"name": [
142-
"John",
143-
"Dan",
144-
"James",
145-
"Amy",
146-
"Anne",
147-
"Ron",
148-
"Alex",
149-
"Bob",
150-
"Charles",
151-
"David",
152-
"Edward",
153-
],
154-
"managerId": [None, 101, 101, 101, 101, 101, 110, 110, 110, 110, 110],
155-
},
156-
{"name": ["John", "David"]},
157-
id="happy_path_two_with_five",
158-
),
159-
pytest.param(
160-
{
161-
"id": [101, 102, 103, 104, 105, 106],
162-
"name": ["John", "Dan", "James", "Amy", "Anne", "Ron"],
163-
"managerId": [None, 101, 101, 101, 101, 102],
164-
},
165-
{"name": []},
166-
id="none_with_five",
167-
),
168-
],
85+
PARAMS_PROBLEM_570,
16986
)
17087
def test_problem_570(input_data, expected_data):
17188
table = pa.Table.from_pydict(input_data)

0 commit comments

Comments
 (0)