Future proofs dtype constraint tests, improves example test checks

Author: pranavm-nvidia

tripy/tests/test_examples.py

@@ -93,8 +93,9 @@ def __str__(self):
 
 
 def test_all_examples_tested():
-    num_dirs = len([d for d in os.listdir(EXAMPLES_ROOT) if os.path.isdir(os.path.join(EXAMPLES_ROOT, d))])
-    assert num_dirs == len(EXAMPLES), "Not all examples are being tested!"
+    tested_examples = set(os.path.relpath(example.path, EXAMPLES_ROOT) for example in EXAMPLES)
+    all_examples = set(d for d in os.listdir(EXAMPLES_ROOT) if os.path.isdir(os.path.join(EXAMPLES_ROOT, d)))
+    assert tested_examples == all_examples, "Not all examples are being tested!"
 
 
 # We want to test our examples with both the latest commit and public build.

tripy/tests/utils/wrappers/test_datatype_constraints.py

@@ -222,8 +222,7 @@ def test_datatype_constraints(case: DtypeConstraintCase):
             outputs = make_list(case.func(**inputs))
 
             # Some APIs do not generate Tensor outputs (e.g. `allclose`), so we don't need to evaluate those.
-            # For DimensionSizes, we don't need to check the type either (they can only be one type)
-            if not any(type(out) == tp.Tensor for out in outputs):
+            if not any(isinstance(out, tp.Tensor) for out in outputs):
                 return
 
             expected_return_types = [