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