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'])