From ba37c47a4dec552d23a446a3aca4e16252f4f6cf Mon Sep 17 00:00:00 2001 From: Johnnie Gray Date: Fri, 15 Sep 2023 10:27:31 -0700 Subject: [PATCH] add more test cases --- Cargo.lock | 2 +- tests/test_cotengrust.py | 9 +++++++++ 2 files changed, 10 insertions(+), 1 deletion(-) diff --git a/Cargo.lock b/Cargo.lock index 6aa53ae..9b2800d 100755 --- a/Cargo.lock +++ b/Cargo.lock @@ -37,7 +37,7 @@ checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd" [[package]] name = "cotengrust" -version = "0.1.1" +version = "0.1.2" dependencies = [ "bit-set", "ordered-float", diff --git a/tests/test_cotengrust.py b/tests/test_cotengrust.py index 7357fb7..db386ac 100644 --- 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",