Add representation tests when pref_vector is not None (#233)

Author: ValerianRey

tests/unit/aggregation/test_aligned_mtl.py

@@ -1,3 +1,4 @@
+import torch
 from pytest import mark
 
 from torchjd.aggregation import AlignedMTL
@@ -14,3 +15,7 @@ def test_representations():
     A = AlignedMTL(pref_vector=None)
     assert repr(A) == "AlignedMTL(pref_vector=None)"
     assert str(A) == "AlignedMTL"
+
+    A = AlignedMTL(pref_vector=torch.tensor([1.0, 2.0, 3.0], device="cpu"))
+    assert repr(A) == "AlignedMTL(pref_vector=tensor([1., 2., 3.]))"
+    assert str(A) == "AlignedMTL([1., 2., 3.])"

tests/unit/aggregation/test_dualproj.py

@@ -1,3 +1,4 @@
+import torch
 from pytest import mark
 
 from torchjd.aggregation import DualProj
@@ -22,3 +23,15 @@ def test_representations():
         repr(A) == "DualProj(pref_vector=None, norm_eps=0.0001, reg_eps=0.0001, solver='quadprog')"
     )
     assert str(A) == "DualProj"
+
+    A = DualProj(
+        pref_vector=torch.tensor([1.0, 2.0, 3.0], device="cpu"),
+        norm_eps=0.0001,
+        reg_eps=0.0001,
+        solver="quadprog",
+    )
+    assert (
+        repr(A) == "DualProj(pref_vector=tensor([1., 2., 3.]), norm_eps=0.0001, reg_eps=0.0001, "
+        "solver='quadprog')"
+    )
+    assert str(A) == "DualProj([1., 2., 3.])"

tests/unit/aggregation/test_upgrad.py

@@ -47,3 +47,15 @@ def test_representations():
     A = UPGrad(pref_vector=None, norm_eps=0.0001, reg_eps=0.0001, solver="quadprog")
     assert repr(A) == "UPGrad(pref_vector=None, norm_eps=0.0001, reg_eps=0.0001, solver='quadprog')"
     assert str(A) == "UPGrad"
+
+    A = UPGrad(
+        pref_vector=torch.tensor([1.0, 2.0, 3.0], device="cpu"),
+        norm_eps=0.0001,
+        reg_eps=0.0001,
+        solver="quadprog",
+    )
+    assert (
+        repr(A) == "UPGrad(pref_vector=tensor([1., 2., 3.]), norm_eps=0.0001, reg_eps=0.0001, "
+        "solver='quadprog')"
+    )
+    assert str(A) == "UPGrad([1., 2., 3.])"