mirror of https://github.com/exaloop/codon
121 lines
4.4 KiB
Python
121 lines
4.4 KiB
Python
|
# 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)
|