Skip to content

Commit 313d966

Browse files
committed
test: deduplicate pytest params for problem 1484
1 parent 656924e commit 313d966

File tree

3 files changed

+41
-76
lines changed

3 files changed

+41
-76
lines changed

tests/test_datafusion.py

Lines changed: 1 addition & 38 deletions
Original file line numberDiff line numberDiff line change
@@ -152,44 +152,7 @@ def test_problem_1378(input_data_1, input_data_2, expected_data):
152152

153153
@pytest.mark.parametrize(
154154
"input_data, expected_data",
155-
[
156-
pytest.param(
157-
{
158-
"sell_date": [
159-
datetime(2020, 5, 30),
160-
datetime(2020, 6, 1),
161-
datetime(2020, 6, 2),
162-
datetime(2020, 5, 30),
163-
datetime(2020, 6, 1),
164-
datetime(2020, 6, 2),
165-
datetime(2020, 5, 30),
166-
],
167-
"product": [
168-
"Headphone",
169-
"Pencil",
170-
"Mask",
171-
"Basketball",
172-
"Bible",
173-
"Mask",
174-
"T-Shirt",
175-
],
176-
},
177-
{
178-
"sell_date": [
179-
datetime(2020, 5, 30),
180-
datetime(2020, 6, 1),
181-
datetime(2020, 6, 2),
182-
],
183-
"num_sold": [3, 2, 1],
184-
"products": [
185-
"Basketball,Headphone,T-Shirt",
186-
"Bible,Pencil",
187-
"Mask",
188-
],
189-
},
190-
id="happy_path",
191-
),
192-
],
155+
PARAMS_PROBLEM_1484,
193156
)
194157
def test_problem_1484(input_data, expected_data):
195158
table = pa.Table.from_pydict(input_data)

tests/test_pandas.py

Lines changed: 1 addition & 38 deletions
Original file line numberDiff line numberDiff line change
@@ -642,44 +642,7 @@ def test_problem_1378(input_data_1, input_data_2, expected_data):
642642

643643
@pytest.mark.parametrize(
644644
"input_data, expected_data",
645-
[
646-
pytest.param(
647-
{
648-
"sell_date": [
649-
datetime(2020, 5, 30),
650-
datetime(2020, 6, 1),
651-
datetime(2020, 6, 2),
652-
datetime(2020, 5, 30),
653-
datetime(2020, 6, 1),
654-
datetime(2020, 6, 2),
655-
datetime(2020, 5, 30),
656-
],
657-
"product": [
658-
"Headphone",
659-
"Pencil",
660-
"Mask",
661-
"Basketball",
662-
"Bible",
663-
"Mask",
664-
"T-Shirt",
665-
],
666-
},
667-
{
668-
"sell_date": [
669-
datetime(2020, 5, 30),
670-
datetime(2020, 6, 1),
671-
datetime(2020, 6, 2),
672-
],
673-
"num_sold": [3, 2, 1],
674-
"products": [
675-
"Basketball,Headphone,T-Shirt",
676-
"Bible,Pencil",
677-
"Mask",
678-
],
679-
},
680-
id="happy_path",
681-
),
682-
],
645+
PARAMS_PROBLEM_1484,
683646
)
684647
def test_problem_1484(input_data, expected_data):
685648
table = pd.DataFrame(input_data)

tests/test_problem_params.py

Lines changed: 39 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1319,6 +1319,45 @@
13191319
]
13201320

13211321

1322+
PARAMS_PROBLEM_1484 = [
1323+
pytest.param(
1324+
{
1325+
"sell_date": [
1326+
datetime(2020, 5, 30),
1327+
datetime(2020, 6, 1),
1328+
datetime(2020, 6, 2),
1329+
datetime(2020, 5, 30),
1330+
datetime(2020, 6, 1),
1331+
datetime(2020, 6, 2),
1332+
datetime(2020, 5, 30),
1333+
],
1334+
"product": [
1335+
"Headphone",
1336+
"Pencil",
1337+
"Mask",
1338+
"Basketball",
1339+
"Bible",
1340+
"Mask",
1341+
"T-Shirt",
1342+
],
1343+
},
1344+
{
1345+
"sell_date": [
1346+
datetime(2020, 5, 30),
1347+
datetime(2020, 6, 1),
1348+
datetime(2020, 6, 2),
1349+
],
1350+
"num_sold": [3, 2, 1],
1351+
"products": [
1352+
"Basketball,Headphone,T-Shirt",
1353+
"Bible,Pencil",
1354+
"Mask",
1355+
],
1356+
},
1357+
id="happy_path",
1358+
),
1359+
]
1360+
13221361
PARAMS_PROBLEM_1517 = [
13231362
pytest.param(
13241363
{

0 commit comments

Comments
 (0)