def test_get_max_partition_from_mal_valid_part_names(self):
max_partition = \
HiveMetastoreHook._get_max_partition_from_part_names(['some_key=value1',
'some_key=value2',
'some_key=value3'],
'some_key')
self.assertEqual(max_partition, 'value3')
def test_get_max_partition_from_mal_formatted_part_names(self):
with self.assertRaises(AirflowException):
HiveMetastoreHook._get_max_partition_from_part_names(
['bad_partition_name'], 'some_key')
def test_get_max_partition_from_empty_part_names(self):
max_partition = \
HiveMetastoreHook._get_max_partition_from_part_names([], 'some_key')
self.assertIsNone(max_partition)
