def test_check_data_str(): with pytest.raises(Exception) as e_info: helpers.check_data(str(1))
def test_check_data_df_list(): helpers.check_data([pd.DataFrame([0, 1, 2]), pd.DataFrame([0, 1, 2])]) == 'dflist'
def test_check_data_df(): helpers.check_data(pd.DataFrame([0, 1, 2])) == 'df'
def test_check_data_array(): helpers.check_data(np.array([[0, 1, 2], [1, 2, 3]])) == 'array'
def test_check_data_list_of_other(): with pytest.raises(ValueError) as e_info: helpers.check_data([1, 2, 3])
def test_check_data_list_of_arrays(): helpers.check_data([np.random.random((3, 3))] * 2) == 'list'