def test_presto_where_latest_partition(self): db = mock.Mock() db.get_indexes = mock.Mock(return_value=[{"column_names": ["ds", "hour"]}]) db.get_extra = mock.Mock(return_value={}) df = pd.DataFrame({"ds": ["01-01-19"], "hour": [1]}) db.get_df = mock.Mock(return_value=df) columns = [{"name": "ds"}, {"name": "hour"}] result = PrestoEngineSpec.where_latest_partition( "test_table", "test_schema", db, select(), columns ) query_result = str(result.compile(compile_kwargs={"literal_binds": True})) self.assertEqual("SELECT \nWHERE ds = '01-01-19' AND hour = 1", query_result)
def test_presto_where_latest_partition(self): db = mock.Mock() db.get_indexes = mock.Mock(return_value=[{ 'column_names': ['ds', 'hour'] }]) df = pd.DataFrame({'ds': ['01-01-19'], 'hour': [1]}) db.get_df = mock.Mock(return_value=df) columns = [{'name': 'ds'}, {'name': 'hour'}] result = PrestoEngineSpec.where_latest_partition( 'test_table', 'test_schema', db, select(), columns) query_result = str( result.compile(compile_kwargs={'literal_binds': True})) self.assertEqual('SELECT \nWHERE ds = \'01-01-19\' AND hour = 1', query_result)