# Simplified version of Python's unittest.TestCase to allow
# copy/pasting tests directly from CPython's test suite.

class TestCase:
    def fail(self, standard_message: str, special_message: str = ''):
        print('TEST FAILED:', special_message if special_message else standard_message)

    def assertTrue(self, obj, msg = ''):
        if not bool(obj):
            self.fail(f'expected object to be true: {obj}', msg)

    def assertFalse(self, obj, msg = ''):
        if bool(obj):
            self.fail(f'expected object to be false: {obj}', msg)

    def assertEqual(self, first, second, msg = ''):
        result = (first == second)
        if not result:
            self.fail(f'expected equality of:\n  1: {first}\n  2: {second}', msg)

    def assertNotEqual(self, first, second, msg = ''):
        result = (first != second)
        if not result:
            self.fail(f'expected inequality of:\n  1: {first}\n  2: {second}', msg)

    def assertSequenceEqual(self, seq1, seq2, msg = ''):
        len1 = len(seq1)
        len2 = len(seq2)
        if len1 != len2:
            self.fail(f'expected equality of sequences (len1={len1}, len2={len2}):\n  1: {seq1}\n  2: {seq2}', msg)

        for i in range(len1):
            a, b = seq1[i], seq2[i]
            if a != b:
                self.fail(f'expected equality of sequences (diff at elem {i}):\n  1: {seq1}\n  2: {seq2}', msg)

    def assertIn(self, member, container, msg = ''):
        if member not in container:
            self.fail(f'expected {member} to be in {container}', msg)

    def assertNotIn(self, member, container, msg = ''):
        if member in container:
            self.fail(f'expected {member} to not be in {container}', msg)

    def assertIs(self, expr1, expr2, msg = ''):
        if expr1 is not expr2:
            self.fail(f'expected {expr1} to be identical to {expr2}', msg)

    def assertIsNot(self, expr1, expr2, msg = ''):
        if expr1 is expr2:
            self.fail(f'expected {expr1} to not be identical to {expr2}', msg)

    def assertIsNot(self, expr1, expr2, msg = ''):
        if expr1 is expr2:
            self.fail(f'expected {expr1} to not be identical to {expr2}', msg)

    def assertCountEqual(self, first, second, msg = ''):
        from collections import Counter
        first_seq, second_seq = list(first), list(second)

        first_counter = Counter(first_seq)
        second_counter = Counter(second_seq)

        if first_counter != second_counter:
            self.fail(f'expected equal counts:\n  1: {first}\n  2: {second}', msg)

    def assertLess(self, a, b, msg = ''):
        if not (a < b):
            self.fail(f'expected less-than:\n  1: {a}\n  2: {b}', msg)

    def assertLessEqual(self, a, b, msg = ''):
        if not (a <= b):
            self.fail(f'expected less-than-or-equal:\n  1: {a}\n  2: {b}', msg)

    def assertGreater(self, a, b, msg = ''):
        if not (a > b):
            self.fail(f'expected greater-than:\n  1: {a}\n  2: {b}', msg)

    def assertGreaterEqual(self, a, b, msg = ''):
        if not (a >= b):
            self.fail(f'expected greater-than-or-equal:\n  1: {a}\n  2: {b}', msg)

    def assertIsNone(self, obj, msg = ''):
        if obj is not None:
            self.fail(f'expected {obj} to be None', msg)

    def assertIsNotNone(self, obj, msg = ''):
        if obj is None:
            self.fail(f'expected {obj} to not be None', msg)

    def assertRaises(self, exception: type, function, *args, **kwargs):
        try:
            function(*args, **kwargs)
        except exception:
            return
        except:
            pass
        self.fail(f'call to function did not raise the given exception')

    def assertAlmostEqual(self, first, second, places: int = 0, msg= '',
                          delta = None):
        if first == second:
            # shortcut
            return
        if places <= 0 and delta is None:
            raise ValueError('specify delta or places not both')

        standard_msg = ''
        diff = abs(first - second)
        if delta is not None:
            if diff <= delta:
                return
            standard_msg = f'{first} != {second} within {delta} delta ({diff} difference)'
        else:
            if places <= 0:
                places = 7
            if round(diff, places) == 0:
                return
            standard_msg = f'{first} != {second} within {places} places ({diff} difference)'
        self.fail(standard_msg, msg)