def test_query_with_date_interval(self):
        task = LastCountryOfUserPartitionTask(**self._get_kwargs())
        query = task.query()
        expected_query = textwrap.dedent("""
            USE default;

            ALTER TABLE last_country_of_user ADD IF NOT EXISTS PARTITION (dt='2014-01-01');
            """)
        self.assertEquals(query, expected_query)
    def test_query_with_date_interval(self):
        task = LastCountryOfUserPartitionTask(**self._get_kwargs())
        query = task.query()
        expected_query = textwrap.dedent(
            """
            USE default;

            ALTER TABLE last_country_of_user ADD IF NOT EXISTS PARTITION (dt='2014-01-01');
            """
        )
        self.assertEquals(query, expected_query)