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)
