add more test cases

jcmgray · Sep 15, 2023 · ba37c47 · ba37c47
1 parent 722b352
commit ba37c47
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/tests/test_cotengrust.py b/tests/test_cotengrust.py
@@ -56,6 +56,15 @@ def get_rand_size_dict(inputs, d_min=2, d_max=3):
 
 # these are taken from opt_einsum
 test_case_eqs = [
+    # Test single-term equations
+    "->",
+    "a->a",
+    "ab->ab",
+    "ab->ba",
+    "abc->bca",
+    "abc->b",
+    "baa->ba",
+    "aba->b",
     # Test scalar-like operations
     "a,->a",
     "ab,->ab",