def read_written_data(context):
# pylint: disable=unused-argument
data_set = SQLTableDataSet(table_name=TABLE_B,
credentials=dict(con=_get_test_sql_con()))
result = data_set.load()
assert result.equals(_get_sample_data())
def prepare_sql_query(context):
sample_data = _get_sample_data()
table_data_set = SQLTableDataSet(table_name=TABLE_C,
credentials=dict(con=_get_test_sql_con()))
table_data_set.save(sample_data)
sql = "SELECT * FROM C WHERE name='me'"
context.query_data_set = SQLQueryDataSet(
sql=sql, credentials=dict(con=_get_test_sql_con()))
def test_empty_connection(self):
"""Check the error when instantiating with an empty
connection string"""
pattern = (r"`con` argument cannot be empty\. "
r"Please provide a SQLAlchemy connection string\.")
with pytest.raises(DataSetError, match=pattern):
SQLTableDataSet(table_name=TABLE_NAME, credentials=dict(con=""))
def test_empty_table_name(self):
"""Check the error when instantiating with an empty table"""
pattern = r"`table\_name` argument cannot be empty\."
with pytest.raises(DataSetError, match=pattern):
SQLTableDataSet(table_name="", credentials=dict(con=CONNECTION))
def table_data_set(request):
kwargs = dict(table_name=TABLE_NAME, credentials=dict(con=CONNECTION))
kwargs.update(request.param)
return SQLTableDataSet(**kwargs)
def prepare_sql_data(context):
_prepare_read_samples()
context.data_set_a = SQLTableDataSet(
table_name=TABLE_A, credentials=dict(con=_get_test_sql_con()))
context.data_set_b = SQLTableDataSet(
table_name=TABLE_B, credentials=dict(con=_get_test_sql_con()))
def _prepare_read_samples():
_clear_db_()
data_set = SQLTableDataSet(table_name=TABLE_A,
credentials=dict(con=_get_test_sql_con()))
data_set.save(_get_sample_data())
