Improve test formatting

Author: ValerianRey

tests/unit/autojac/_transform/test_accumulate.py

@@ -102,5 +102,4 @@ def test_check_keys():
     accumulate = Accumulate()
 
     output_keys = accumulate.check_keys({key})
-
     assert output_keys == set()

tests/unit/autojac/_transform/test_aggregate.py

@@ -161,7 +161,6 @@ def test_aggregate_matrices_check_keys():
     aggregate = _AggregateMatrices(Random(), [key2, key1])
 
     output_keys = aggregate.check_keys({key1, key2})
-
     assert output_keys == {key1, key2}
 
     with raises(RequirementError):
@@ -179,7 +178,6 @@ def test_matrixify_check_keys():
     matrixify = _Matrixify()
 
     output_keys = matrixify.check_keys({key1, key2})
-
     assert output_keys == {key1, key2}
 
 
@@ -191,5 +189,4 @@ def test_reshape_check_keys():
     reshape = _Reshape()
 
     output_keys = reshape.check_keys({key1, key2})
-
     assert output_keys == {key1, key2}

tests/unit/autojac/_transform/test_base.py

@@ -45,7 +45,6 @@ def test_composition_check_keys():
     t2 = FakeTransform(required_keys={a2}, output_keys={a1})
 
     output_keys = (t1 << t2).check_keys({a2})
-
     assert output_keys == {a1, a2}
 
     # Inner Transform fails its check
@@ -71,7 +70,6 @@ def test_conjunct_check_keys_1():
     t3 = FakeTransform(required_keys={a2}, output_keys=set())
 
     output_keys = (t1 | t2).check_keys({a1})
-
     assert output_keys == set()
 
     with raises(RequirementError):
@@ -95,7 +93,6 @@ def test_conjunct_check_keys_2():
     t3 = FakeTransform(required_keys=set(), output_keys={a2})
 
     output_keys = (t2 | t3).check_keys(set())
-
     assert output_keys == {a1, a2}
 
     with raises(RequirementError):

tests/unit/autojac/_transform/test_diagonalize.py

@@ -109,7 +109,6 @@ def test_check_keys():
     diag = Diagonalize([key1])
 
     output_keys = diag.check_keys({key1})
-
     assert output_keys == {key1}
 
     with raises(RequirementError):

tests/unit/autojac/_transform/test_grad.py

@@ -297,7 +297,6 @@ def test_check_keys():
     grad = Grad(outputs=[y], inputs=[a1, a2])
 
     output_keys = grad.check_keys({y})
-
     assert output_keys == {a1, a2}
 
     with raises(RequirementError):

tests/unit/autojac/_transform/test_init.py

@@ -71,7 +71,6 @@ def test_check_keys():
     init = Init([key])
 
     output_keys = init.check_keys(set())
-
     assert output_keys == {key}
 
     with raises(RequirementError):

tests/unit/autojac/_transform/test_jac.py

@@ -297,7 +297,6 @@ def test_check_keys():
     jac = Jac(outputs=[y], inputs=[a1, a2], chunk_size=None)
 
     output_keys = jac.check_keys({y})
-
     assert output_keys == {a1, a2}
 
     with raises(RequirementError):

tests/unit/autojac/_transform/test_select.py

@@ -67,7 +67,6 @@ def test_check_keys():
     key3 = torch.tensor([3.0])
 
     output_keys = Select([key1, key2]).check_keys({key1, key2, key3})
-
     assert output_keys == {key1, key2}
 
     with raises(RequirementError):

tests/unit/autojac/test_backward.py

@@ -25,7 +25,6 @@ def test_check_create_transform():
         parallel_chunk_size=None,
     )
     output_keys = transform.check_keys(set())
-
     assert output_keys == set()
 
 

tests/unit/autojac/test_mtl_backward.py

@@ -29,8 +29,8 @@ def test_check_create_transform():
         retain_graph=False,
         parallel_chunk_size=None,
     )
-    output_keys = transform.check_keys(set())
 
+    output_keys = transform.check_keys(set())
     assert output_keys == set()