Override assertEqual only in test cases (#2513) (#2526)

* Override assert equal only in test cases
* Support python -m pytest

---------

Co-authored-by: Jinghui <jinghui.gu@intel.com>

Author: cchheennhhaaoo

intel_extension_for_pytorch/xpu/__init__.py

@@ -496,4 +496,7 @@ def get_device_type() -> str:
 if intel_extension_for_pytorch._C._has_xpu():
     if is_available() and not has_fp64_dtype():
         override_tensor_totype()
-        override_assert_equal()
+
+        exec_path = sys.argv[0].split("/")
+        if (len(exec_path) > 0 and "pytest" in exec_path):
+            override_assert_equal()