Fixup for tests details

[modules/adao.git] / test / test6702 / Doc_TUI_Exemple_02.py

diff --git a/test/test6702/Doc_TUI_Exemple_02.py b/test/test6702/Doc_TUI_Exemple_02.py
index 0877647c3494e44a93647c17604bf11d817017e8..e6af4aca23d1078aebd2f4f6bb146309b97b78af 100644 (file)
--- a/test/test6702/Doc_TUI_Exemple_02.py
+++ b/test/test6702/Doc_TUI_Exemple_02.py
@@ -21,8 +21,6 @@
 # Author: Jean-Philippe Argaud, jean-philippe.argaud@edf.fr, EDF R&D
 "Verification d'un exemple de la documentation"
 
-from utExtend import assertAlmostEqualArrays
-
 # ==============================================================================
 def test1():
     """Test"""
@@ -65,6 +63,20 @@ def test2():
     #
     return case.get("Analysis")[-1]
 
+# ==============================================================================
+def assertAlmostEqualArrays(first, second, places=7, msg=None, delta=None):
+    "Compare two vectors, like unittest.assertAlmostEqual"
+    import numpy
+    if msg is not None:
+        print(msg)
+    if delta is not None:
+        if ( (numpy.asarray(first) - numpy.asarray(second)) > float(delta) ).any():
+            raise AssertionError("%s != %s within %s places"%(first,second,delta))
+    else:
+        if ( (numpy.asarray(first) - numpy.asarray(second)) > 10**(-int(places)) ).any():
+            raise AssertionError("%s != %s within %i places"%(first,second,places))
+    return max(abs(numpy.asarray(first) - numpy.asarray(second)))
+
 # ==============================================================================
 if __name__ == "__main__":
     print('\nAUTODIAGNOSTIC\n')