def test_get_target_partitions_string(self): """ This method tests the correct functionality of get_partition_column_string of Util class :return: """ assert Util.get_target_partitions_string("year") == "year" assert Util.get_target_partitions_string("month") == "year,month" assert Util.get_target_partitions_string("day") == "year,month,day" assert Util.get_target_partitions_string("") == "" assert Util.get_target_partitions_string("country") == "country"
def test_get_target_partitions_string(self): """ This method tests the correct functionality of get_partition_column_string of Util class :return: """ assert Util.get_target_partitions_string("year") == "year" assert Util.get_target_partitions_string("month") == "year,month" assert Util.get_target_partitions_string("day") == "year,month,day" assert Util.get_target_partitions_string("") == "" with pytest.raises(Exception) as exc_info: Util.get_target_partitions_list("country") assert "Partition type country not supported" in str(exc_info.value)