def test_hex_values(self): # Test hex values self.failUnlessEqual('#000000', _get_color('!000000', 'BG')) self.failUnlessEqual('#FFFFFF', _get_color('!ffffff', 'BG')) def test_short_hex_values(self): # Shorts self.failUnlessEqual('#001122', _get_color('!012', 'BG')) self.failUnlessEqual('#AABBCC', _get_color('!abc', 'BG')) def test_invalid_hex_values(self): # invalids self.failUnlessEqual('grey', _get_color('!xyz', 'TEXT')) self.failUnlessEqual('grey', _get_color('!1234567', 'BORDER')) def test_invalid_type(self): # Test exceptions # invalid type self.failUnlessRaises(KeyError, _get_color, '', 'TYPE') self.failUnlessRaises(KeyError, _get_color, '!invalid', 'TYPE') if __name__ == '__main__': import os from django.core.management.commands.test import Command test_argv = [os.sys.argv[0], '', 'dummyimage'] + os.sys.argv[1:] test_command = Command() test_command.run_from_argv(test_argv)