def test_max_field_len_parameter(test_dataframe):
"""
Test.
"""
with pytest.raises(AssertionError):
sample_row(test_dataframe, random_state="a")
with pytest.raises(AssertionError):
sample_row(test_dataframe, random_state=None)
def test_filter_rows_with_na_parameter(na_col_dataframe):
"""
Test.
"""
with pytest.raises(AssertionError):
sample_row(test_dataframe, filter_rows_with_na="a")
with pytest.raises(AssertionError):
sample_row(test_dataframe, filter_rows_with_na=None)
def test_df_with_nan_in_row(na_in_row_dataframe, expected_na_in_row_return):
pd.testing.assert_frame_equal(
sample_row(na_in_row_dataframe,
random_state=42,
filter_rows_with_na=True),
expected_na_in_row_return,
)
def test_df_with_nan_column(na_col_dataframe, expected_na_col_return_w_filtering, caplog):
"""
Test.
"""
with caplog.at_level("INFO"):
pd.testing.assert_frame_equal(
sample_row(na_col_dataframe, random_state=42, filter_rows_with_na=True),
expected_na_col_return_w_filtering,
)
assert "No rows without NaN found, sampling from all rows.." in caplog.text
def test_long_str_df(long_str_dataframe, expected_long_str_return):
"""
Test.
"""
pd.testing.assert_frame_equal(sample_row(long_str_dataframe, random_state=42), expected_long_str_return)
def test_empty_df(empty_dataframe):
"""
Test.
"""
with pytest.raises(AssertionError):
sample_row(empty_dataframe)
def test_normal_df(test_dataframe, expected_test_return):
"""
Test.
"""
pd.testing.assert_frame_equal(sample_row(test_dataframe), expected_test_return)
def test_random_state_parameter(test_dataframe):
with pytest.raises(AssertionError):
sample_row(test_dataframe, random_state="a")
with pytest.raises(AssertionError):
sample_row(test_dataframe, random_state=None)
