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())