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)