mirror of
https://github.com/triqs/dft_tools
synced 2024-12-26 22:33:48 +01:00
60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
r"""
|
|
Module defining a custom TestCase with extra functionality.
|
|
"""
|
|
|
|
import unittest
|
|
import numpy as np
|
|
import difflib
|
|
|
|
class MyTestCase(unittest.TestCase):
|
|
"""
|
|
Custom TestCase class supporting additional equality checks:
|
|
- numpy array equality
|
|
- file equality
|
|
"""
|
|
def __init__(self, *args, **kwargs):
|
|
"""
|
|
Initializes a custom equality function for comparing numpy arrays.
|
|
"""
|
|
super(MyTestCase, self).__init__(*args, **kwargs)
|
|
np.set_printoptions(suppress=True)
|
|
self.addTypeEqualityFunc(np.ndarray, self.is_arrays_equal)
|
|
|
|
def is_arrays_equal(self, arr1, arr2, msg=None):
|
|
"""
|
|
Raises self.failureException is arrays arr1 and arr2
|
|
are not equal.
|
|
"""
|
|
if not np.allclose(arr1, arr2):
|
|
raise self.failureException(msg)
|
|
|
|
def assertFileEqual(self, file1, file2):
|
|
"""
|
|
Compares two files using difflib.
|
|
Empty lines are ignored.
|
|
Files are assumed to be relatively small;
|
|
the data is truncated for files larger than MAX_SIZE bytes.
|
|
"""
|
|
MAX_SIZE = 100000
|
|
with open(file1, 'r') as f1:
|
|
str1 = f1.read(MAX_SIZE)
|
|
with open(file2, 'r') as f2:
|
|
str2 = f2.read(MAX_SIZE)
|
|
#
|
|
# Make a diff
|
|
#
|
|
# Remove empty lines
|
|
lstr1 = filter(lambda s: s.strip() != '', str1.splitlines(True))
|
|
lstr1 = [str1.replace(" ","") for str1 in lstr1] # Remove spaces
|
|
lstr2 = filter(lambda s: s.strip() != '', str2.splitlines(True))
|
|
lstr2 = [str2.replace(" ","") for str2 in lstr2] # Remove spaces
|
|
# diff
|
|
delta = difflib.unified_diff(lstr1, lstr2)
|
|
# combine delta's to a string
|
|
diff = ''.join(delta)
|
|
# if 'diff' is non-empty, files are different
|
|
if diff:
|
|
return self.fail("Files '%s' and '%s' differ"%(file1, file2))
|
|
|
|
|