예제 #1
0
    def test_test_non_decorated_formatter(self):
        formatter = OutputFormatter(False)

        self.assertTrue(formatter.has_style('error'))
        self.assertTrue(formatter.has_style('info'))
        self.assertTrue(formatter.has_style('comment'))
        self.assertTrue(formatter.has_style('question'))

        self.assertEqual('some error',
                         formatter.format('<error>some error</error>'))
        self.assertEqual('some info',
                         formatter.format('<info>some info</info>'))
        self.assertEqual('some comment',
                         formatter.format('<comment>some comment</comment>'))
        self.assertEqual(
            'some question',
            formatter.format('<question>some question</question>'))

        formatter.set_decorated(True)

        self.assertEqual('\033[37;41msome error\033[0m',
                         formatter.format('<error>some error</error>'))
        self.assertEqual('\033[32msome info\033[0m',
                         formatter.format('<info>some info</info>'))
        self.assertEqual('\033[33msome comment\033[0m',
                         formatter.format('<comment>some comment</comment>'))
        self.assertEqual(
            '\033[30;46msome question\033[0m',
            formatter.format('<question>some question</question>'))
예제 #2
0
    def test_test_non_decorated_formatter(self):
        formatter = OutputFormatter(False)

        self.assertTrue(formatter.has_style('error'))
        self.assertTrue(formatter.has_style('info'))
        self.assertTrue(formatter.has_style('comment'))
        self.assertTrue(formatter.has_style('question'))

        self.assertEqual(
            'some error',
            formatter.format('<error>some error</error>')
        )
        self.assertEqual(
            'some info',
            formatter.format('<info>some info</info>')
        )
        self.assertEqual(
            'some comment',
            formatter.format('<comment>some comment</comment>')
        )
        self.assertEqual(
            'some question',
            formatter.format('<question>some question</question>')
        )

        formatter.set_decorated(True)

        self.assertEqual(
            '\033[37;41msome error\033[0m',
            formatter.format('<error>some error</error>')
        )
        self.assertEqual(
            '\033[32msome info\033[0m',
            formatter.format('<info>some info</info>')
        )
        self.assertEqual(
            '\033[33msome comment\033[0m',
            formatter.format('<comment>some comment</comment>')
        )
        self.assertEqual(
            '\033[30;46msome question\033[0m',
            formatter.format('<question>some question</question>')
        )