def testEmptyClassRetrieval(self): """Checks that disabling RETURN_CLASSES returns empty strings for classes.""" saved_setting = default.RETURN_CLASSES default.RETURN_CLASSES = False class_name = default.CSS_CLASSES.keys()[0] self.assertEqual(insert_classes(class_name), "", "Check that insert classes now returns an empty string.") # Restore setting default.RETURN_CLASSES = saved_setting
def testDefaultClassRetrieval(self): """Checks that default values can be retrieved.""" class_name = default.CSS_CLASSES.keys()[0] self.assertEqual(insert_classes(class_name), default.CSS_CLASSES[class_name], "Check that insert classes returns the correct value from the dictionary.")