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