def test_presto_extra_table_metadata(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)
PrestoEngineSpec.get_create_view = mock.Mock(return_value=None)
result = PrestoEngineSpec.extra_table_metadata(db, "test_table", "test_schema")
self.assertEqual({"ds": "01-01-19", "hour": 1}, result["partitions"]["latest"])
def test_presto_extra_table_metadata(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)
result = PrestoEngineSpec.extra_table_metadata(db, 'test_table',
'test_schema')
self.assertEqual({
'ds': '01-01-19',
'hour': 1
}, result['partitions']['latest'])
