def test_validate_parameters_pass_dataframe(key_cols, param_cols, val_cols):
data = pd.DataFrame({
'a': [1, 2],
'b_start': [0, 5],
'b_end': [5, 10],
'c': [100, 150]
})
validate_parameters(data, key_cols, param_cols, val_cols)
def test_validate_parameters_error_dataframe(key_cols, param_cols, val_cols,
match):
data = pd.DataFrame({
'a': [1, 2],
'b_start': [0, 5],
'b_end': [5, 10],
'c': [100, 150]
})
with pytest.raises(ValueError, match=match):
validate_parameters(data, key_cols, param_cols, val_cols)
def test_validate_parameters_pass_scalar_data(key_cols, param_cols, val_cols):
validate_parameters([1, 2, 3], key_cols, param_cols, val_cols)
def test_validate_parameters_fail_other_data(data):
with pytest.raises(TypeError, match='only allowable types'):
validate_parameters(data, [], [], [])
def test_validate_parameters_error_scalar_data(key_cols, param_cols, val_cols,
match):
with pytest.raises(ValueError, match=match):
validate_parameters([1, 2, 3], key_cols, param_cols, val_cols)
def test_validate_parameters_no_data(data):
with pytest.raises(ValueError, match='supply some data'):
validate_parameters(data, [], [], [])