Skip to content
Original file line number Diff line number Diff line change
Expand Up @@ -62,6 +62,178 @@
},
),
),
(
Table(
{
"col1": [
"A",
"B",
"A",
"outlier_col3",
"a",
"a",
"a",
"a",
"a",
"a",
"outlier_col2",
"a",
],
"col2": [1.0, 2.0, 3.0, 4.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, -1000.0, None],
"col3": [2, 3, 1, 1_000_000_000, 1, 1, 1, 1, 1, 1, 1, 1],
},
),
Table(
{
"col1": [
"A",
"B",
"A",
"a",
"a",
"a",
"a",
"a",
"a",
"a",
],
"col2": [1.0, 2.0, 3.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, None],
"col3": [2, 3, 1, 1, 1, 1, 1, 1, 1, 1],
},
),
),
(
Table(
{
"col1": [
"A",
"B",
"A",
"positive_outlier",
"a",
"a",
"a",
"a",
"a",
"a",
"a",
"a",
"negative_outlier",
"a",
"a",
"a",
"a",
"a",
"a",
"a",
"a",
"a",
"a",
],
"col2": [
1.0,
2.0,
3.0,
4.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
None,
1.0,
2.0,
1.0,
4.0,
1.0,
3.0,
1.0,
2.0,
1.0,
4.0,
1.0,
],
"col3": [
2,
3,
1,
1_000_000_000_000,
1,
1,
1,
1,
1,
1,
1,
1,
-1_000_000_000_000,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
],
},
),
Table(
{
"col1": [
"A",
"B",
"A",
"a",
"a",
"a",
"a",
"a",
"a",
"a",
"a",
"a",
"a",
"a",
"a",
"a",
"a",
"a",
"a",
"a",
"a",
],
"col2": [
1.0,
2.0,
3.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
None,
2.0,
1.0,
4.0,
1.0,
3.0,
1.0,
2.0,
1.0,
4.0,
1.0,
],
"col3": [2, 3, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
},
),
),
(
Table(
{
Expand All @@ -78,7 +250,14 @@
),
(Table(), Table()),
],
ids=["no outliers", "with outliers", "no rows", "empty"],
ids=[
"no outliers",
"one outlier",
"outliers in two different columns",
"multiple outliers in one column",
"no rows",
"empty",
],
)
def test_should_remove_rows_with_outliers(table: Table, expected: Table) -> None:
updated_table = table.remove_rows_with_outliers()
Expand Down