def test_when_sample_is_not_provided_without_turbo(self):
query = Query({
"conditions": [],
"aggregations": [],
"groupby": [],
})
request_settings = RequestSettings(turbo=False,
consistent=False,
debug=False)
clickhouse_query = ClickhouseQuery(
dataset=self.dataset,
query=query,
settings=request_settings,
prewhere_conditions=[],
)
assert 'SAMPLE' not in clickhouse_query.format_sql()
def test_provided_sample_should_be_used_with_turbo(self):
query = Query({
"conditions": [],
"aggregations": [],
"groupby": [],
"sample": 0.1
})
request_settings = RequestSettings(turbo=True,
consistent=False,
debug=False)
clickhouse_query = ClickhouseQuery(
dataset=self.dataset,
query=query,
settings=request_settings,
prewhere_conditions=[],
)
assert 'SAMPLE 0.1' in clickhouse_query.format_sql()
def test_when_sample_is_not_provided_with_turbo(self):
source = self.dataset.get_dataset_schemas().get_read_schema(
).get_data_source()
query = Query(
{
"conditions": [],
"aggregations": [],
"groupby": [],
},
source,
)
request_settings = RequestSettings(turbo=True,
consistent=False,
debug=False)
clickhouse_query = ClickhouseQuery(
dataset=self.dataset,
query=query,
settings=request_settings,
prewhere_conditions=[],
)
assert "SAMPLE 0.2" in clickhouse_query.format_sql()
def test_provided_sample_should_be_used(self):
source = self.dataset.get_dataset_schemas().get_read_schema(
).get_data_source()
query = Query(
{
"conditions": [],
"aggregations": [],
"groupby": [],
"sample": 0.1
},
source,
)
request_settings = RequestSettings(turbo=False,
consistent=False,
debug=False)
clickhouse_query = ClickhouseQuery(
dataset=self.dataset,
query=query,
settings=request_settings,
prewhere_conditions=[],
)
assert 'SAMPLE 0.1' in clickhouse_query.format_sql()