def METHOD_NAME(self):
    """
    Test sample data are returned based on user query
    """
    stmt = "`age` > 30"
    sampler = DatalakeSampler(
        client=FakeConnection().client(),
        table=[self.df1, self.df2],
        profile_sample_query=stmt,
    )
    sample_data = sampler.fetch_sample_data()
    assert len(sample_data.columns) == 8
    assert len(sample_data.rows) == 3