rename tests with check_and_get_keys to check_keys

Author: PierreQuinton

tests/unit/autojac/_transform/test_accumulate.py

@@ -95,8 +95,8 @@ def test_no_leaf_and_no_retains_grad_fails():
         accumulate(input)
 
 
-def test_check_and_get_keys():
-    """Tests that the `check_and_get_keys` method works correctly."""
+def test_check_keys():
+    """Tests that the `check_keys` method works correctly."""
 
     key = torch.tensor([1.0], requires_grad=True)
     accumulate = Accumulate()

tests/unit/autojac/_transform/test_aggregate.py

@@ -144,8 +144,8 @@ def test_reshape():
     assert_tensor_dicts_are_close(output, expected_output)
 
 
-def test_aggregate_matrices_check_and_get_keys():
-    """Tests that the `check_and_get_keys` method works correctly."""
+def test_aggregate_matrices_check_keys():
+    """Tests that the `check_keys` method works correctly."""
 
     key1 = torch.tensor([1.0])
     key2 = torch.tensor([2.0])
@@ -156,8 +156,8 @@ def test_aggregate_matrices_check_and_get_keys():
     assert output_keys == {key1, key2}
 
 
-def test_matrixify_check_and_get_keys():
-    """Tests that the `check_and_get_keys` method works correctly."""
+def test_matrixify_check_keys():
+    """Tests that the `check_keys` method works correctly."""
 
     key1 = torch.tensor([1.0])
     key2 = torch.tensor([2.0])
@@ -168,8 +168,8 @@ def test_matrixify_check_and_get_keys():
     assert output_keys == {key1, key2}
 
 
-def test_reshape_check_and_get_keys():
-    """Tests that the `check_and_get_keys` method works correctly."""
+def test_reshape_check_keys():
+    """Tests that the `check_keys` method works correctly."""
 
     key1 = torch.tensor([1.0])
     key2 = torch.tensor([2.0])

tests/unit/autojac/_transform/test_base.py

@@ -33,9 +33,9 @@ def check_keys(self, input_keys: set[Tensor]) -> set[Tensor]:
         return self._output_keys
 
 
-def test_composition_check_and_get_keys():
+def test_composition_check_keys():
     """
-    Tests that `check_and_get_keys` works correctly for a composition of transforms: the inner
+    Tests that `check_keys` works correctly for a composition of transforms: the inner
     transform's `output_keys` has to match with the outer transform's `required_keys`.
     """
 
@@ -52,9 +52,9 @@ def test_composition_check_and_get_keys():
         (t2 << t1).check_keys({a1, a2})
 
 
-def test_conjunct_check_and_get_keys_1():
+def test_conjunct_check_keys_1():
     """
-    Tests that `check_and_get_keys` works correctly for a conjunction of transforms: all transforms
+    Tests that `check_keys` works correctly for a conjunction of transforms: all transforms
     should successfully check their keys.
     """
 
@@ -76,9 +76,9 @@ def test_conjunct_check_and_get_keys_1():
         (t1 | t2 | t3).check_keys({a1, a2})
 
 
-def test_conjunct_check_and_get_keys_2():
+def test_conjunct_check_keys_2():
     """
-    Tests that `check_and_get_keys` works correctly for a conjunction of transforms: their
+    Tests that `check_keys` works correctly for a conjunction of transforms: their
     `output_keys` should be disjoint.
     """
 

tests/unit/autojac/_transform/test_diagonalize.py

@@ -97,8 +97,8 @@ def test_permute_order():
     assert_tensor_dicts_are_close(output, expected_output)
 
 
-def test_check_and_get_keys():
-    """Tests that the `check_and_get_keys` method works correctly."""
+def test_check_keys():
+    """Tests that the `check_keys` method works correctly."""
 
     key = torch.tensor([1.0])
     diag = Diagonalize([key])

tests/unit/autojac/_transform/test_grad.py

@@ -283,8 +283,8 @@ def test_create_graph():
     assert gradients[a].requires_grad
 
 
-def test_check_and_get_keys():
-    """Tests that the `check_and_get_keys` method works correctly."""
+def test_check_keys():
+    """Tests that the `check_keys` method works correctly."""
 
     x = torch.tensor(5.0)
     a1 = torch.tensor(2.0, requires_grad=True)

tests/unit/autojac/_transform/test_init.py

@@ -63,8 +63,8 @@ def test_conjunction_of_inits_is_init():
     assert_tensor_dicts_are_close(output, expected_output)
 
 
-def test_check_and_get_keys():
-    """Tests that the `check_and_get_keys` method works correctly."""
+def test_check_keys():
+    """Tests that the `check_keys` method works correctly."""
 
     key = torch.tensor([1.0])
     init = Init([key])

tests/unit/autojac/_transform/test_interactions.py

@@ -251,9 +251,9 @@ def test_equivalence_jac_grads():
     assert_close(jac_c, torch.stack([grad_1_c, grad_2_c]))
 
 
-def test_stack_check_and_get_keys():
+def test_stack_check_keys():
     """
-    Tests that the `check_and_get_keys` method works correctly for a stack of transforms: all of
+    Tests that the `check_keys` method works correctly for a stack of transforms: all of
     them should have the same `required_keys`.
     """
 

tests/unit/autojac/_transform/test_jac.py

@@ -283,8 +283,8 @@ def test_create_graph():
     assert jacobians[a2].requires_grad
 
 
-def test_check_and_get_keys():
-    """Tests that the `check_and_get_keys` method works correctly."""
+def test_check_keys():
+    """Tests that the `check_keys` method works correctly."""
 
     x = torch.tensor(5.0)
     a1 = torch.tensor(2.0, requires_grad=True)

tests/unit/autojac/_transform/test_select.py

@@ -56,9 +56,9 @@ def test_conjunction_of_selects_is_select():
     assert_tensor_dicts_are_close(output, expected_output)
 
 
-def test_check_and_get_keys():
+def test_check_keys():
     """
-    Tests that the `check_and_get_keys` method works correctly: the set of keys to select should
+    Tests that the `check_keys` method works correctly: the set of keys to select should
     be a subset of the set of required_keys.
     """