def test_assure_pandas_df_array(expected_df, expected_df_2d):
x = np.array([[1, 2, 3], [2, 3, 4]])
x_df = assure_pandas_df(x)
assert x_df.equals(expected_df_2d)
x = np.array([1, 2, 3])
x_df = assure_pandas_df(x)
assert x_df.equals(expected_df)
def test_assure_pandas_df_types():
"""
Test.
"""
with pytest.raises(TypeError):
assure_pandas_df("Test")
with pytest.raises(TypeError):
assure_pandas_df(5)
def test_assure_pandas_df_array(expected_df, expected_df_2d):
"""
Test.
"""
x = np.array([[1, 2, 3], [2, 3, 4]], dtype="int64")
x_df = assure_pandas_df(x)
pd.testing.assert_frame_equal(x_df, expected_df_2d)
x = np.array([1, 2, 3], dtype="int64")
x_df = assure_pandas_df(x)
pd.testing.assert_frame_equal(x_df, expected_df)
def test_assure_pandas_df_df(expected_df_2d):
"""
Test.
"""
x = pd.DataFrame([[1, 2, 3], [2, 3, 4]])
x_df = assure_pandas_df(x)
pd.testing.assert_frame_equal(x_df, expected_df_2d)
def test_assure_pandas_df_series(expected_df):
"""
Test.
"""
x = pd.Series([1, 2, 3])
x_df = assure_pandas_df(x)
pd.testing.assert_frame_equal(x_df, expected_df)
def test_assure_pandas_df_list_of_lists(expected_df_2d):
"""
Test.
"""
x = [[1, 2, 3], [2, 3, 4]]
x_df = assure_pandas_df(x)
pd.testing.assert_frame_equal(x_df, expected_df_2d)
def test_assure_pandas_df_list(expected_df):
"""
Test.
"""
x = [1, 2, 3]
x_df = assure_pandas_df(x)
pd.testing.assert_frame_equal(x_df, expected_df)
def test_assure_pandas_df_df(expected_df_2d):
"""
Test.
"""
x = pd.DataFrame([[1, 2, 3], [2, 3, 4]])
x_df = assure_pandas_df(x)
assert x_df.equals(expected_df_2d)
def test_assure_pandas_df_series(expected_df):
"""
Test.
"""
x = pd.Series([1, 2, 3])
x_df = assure_pandas_df(x)
assert x_df.equals(expected_df)
def test_assure_pandas_df_list_of_lists(expected_df_2d):
"""
Test.
"""
x = [[1, 2, 3], [2, 3, 4]]
x_df = assure_pandas_df(x)
assert x_df.equals(expected_df_2d)
def test_assure_pandas_df_list(expected_df):
"""
Test.
"""
x = [1, 2, 3]
x_df = assure_pandas_df(x)
assert x_df.equals(expected_df)
