cov

Author: xadupre

_unittests/ut_xrun_doc/test_helpers.py

@@ -29,6 +29,7 @@
     rename_dynamic_dimensions,
     rename_dynamic_expression,
 )
+from onnx_diagnostic.cache_helpers import make_dynamic_cache
 
 TFLOAT = onnx.TensorProto.FLOAT
 
@@ -94,6 +95,32 @@ def test_pretty_onnx(self):
         pretty_onnx(proto.graph)
         pretty_onnx(proto.graph.node[0])
 
+    @hide_stdout()
+    def test_print_pretty_onnx(self):
+        proto = oh.make_model(
+            oh.make_graph(
+                [
+                    oh.make_node("Sigmoid", ["Y"], ["sy"]),
+                    oh.make_node("Mul", ["Y", "sy"], ["ysy"]),
+                    oh.make_node("Mul", ["X", "ysy"], ["final"]),
+                ],
+                "nd",
+                [
+                    oh.make_tensor_value_info("X", TFLOAT, [1, "b", "c"]),
+                    oh.make_tensor_value_info("Y", TFLOAT, ["a", "b", "c"]),
+                ],
+                [oh.make_tensor_value_info("final", TFLOAT, ["a", "b", "c"])],
+            ),
+            opset_imports=[oh.make_opsetid("", 18)],
+            ir_version=9,
+        )
+        self.print_onnx(proto)
+        self.print_model(proto)
+        self.dump_onnx("test_print_pretty_onnx", proto)
+        self.check_ort(proto)
+        self.assertNotEmpty(proto)
+        self.assertEmpty(None)
+
     def test_get_onnx_signature(self):
         proto = oh.make_model(
             oh.make_graph(
@@ -121,6 +148,12 @@ def test_flatten(self):
             [
                 torch.rand((5, 6), dtype=torch.float16),
                 torch.rand((5, 6, 7), dtype=torch.float16),
+                {
+                    "a": torch.rand((2,), dtype=torch.float16),
+                    "cache": make_dynamic_cache(
+                        [(torch.rand((4, 4, 4)), torch.rand((4, 4, 4)))]
+                    ),
+                },
             ],
         )
         flat = flatten_object(inputs)

_unittests/ut_xrun_doc/test_unit_test.py

@@ -13,6 +13,8 @@
     is_linux,
     unit_test_going,
     measure_time,
+    has_cuda,
+    has_onnxscript,
 )
 
 
@@ -65,6 +67,8 @@ def test_is(self):
         is_azure()
         is_linux()
         unit_test_going()
+        has_cuda()
+        has_onnxscript()
 
     def test_measure_time(self):
         res = measure_time(lambda: math.cos(0.5))

onnx_diagnostic/ext_test_case.py

@@ -62,24 +62,6 @@ def skipif_ci_apple(msg) -> Callable:
     return lambda x: x
 
 
-def with_path_append(path_to_add: Union[str, List[str]]) -> Callable:
-    """Adds a path to sys.path to check."""
-
-    def wraps(f, path_to_add=path_to_add):
-        def wrapped(self, path_to_add=path_to_add):
-            cpy = sys.path.copy()
-            if path_to_add is not None:
-                if isinstance(path_to_add, str):
-                    path_to_add = [path_to_add]
-                sys.path.extend(path_to_add)
-            f(self)
-            sys.path = cpy
-
-        return wrapped
-
-    return wraps
-
-
 def unit_test_going():
     """
     Enables a flag telling the script is running while testing it.