def test_ascii_image(self):
args1 = parse_args(
['--image', 'test.jpg', '--width', '10', '--height', '5'])
args2 = parse_args(
['--image', 'test.jpg', '--width', '20', '--height', '10'])
self.assertEqual(
create_ascii_image(Image.open(args1.image), int(args1.width),
int(args1.height), GRAYSCALE_10).size,
(110, 85))
self.assertEqual(
create_ascii_image(Image.open(args2.image), int(args2.width),
int(args2.height), GRAYSCALE_40).size,
(220, 170))
def test_args_width_and_height(self):
args1 = parse_args(
['--image', 'test.jpg', '--width', '100', '--height', '50'])
args2 = parse_args(
['--image', 'test.jpg', '--width', '0', '--height', '50'])
args3 = parse_args(
['--image', 'test.jpg', '--width', '100', '--height', '0'])
self.assertEqual(
check_width_and_height(int(args1.width), int(args1.height)), True)
self.assertEqual(
check_width_and_height(int(args2.width), int(args2.height)), False)
self.assertEqual(
check_width_and_height(int(args3.width), int(args3.height)), False)
def test_args_mode(self):
args1 = parse_args([
'--image', 'test.jpg', '--width', '100', '--height', '50',
'--mode', '10'
])
args3 = parse_args([
'--image', 'test.jpg', '--width', '100', '--height', '50',
'--mode', '40'
])
args5 = parse_args([
'--image', 'test.jpg', '--width', '100', '--height', '50',
'--mode', '90'
])
args6 = parse_args(
['--image', 'test.jpg', '--width', '100', '--height', '50'])
self.assertEqual(check_mode(args1), GRAYSCALE_10)
self.assertEqual(check_mode(args3), GRAYSCALE_40)
self.assertFalse(check_mode(args5))
self.assertEqual(check_mode(args6), GRAYSCALE_40)
def test_ansi_check(self):
args1 = parse_args(['--image', '256.ansi'])
args2 = parse_args(
['--image', '256.ansi', '--width', '20', '--height', '10'])
self.assertTrue(check_mode(args1))
self.assertTrue(check_mode(args2))
def test_parser(self):
args = parse_args(
['--image', 'test.jpg', '--width', '100', '--height', '50'])
self.assertTrue(args.image, "test.jpg")
self.assertTrue(args.width, "100")
self.assertTrue(args.height, "50")