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.")
