[Enhancement] Reduce the randomness in unit test of `stylegan3_utils.…

…py` (#1306)

make s3-utils unit test more robustic

tests/test_models/test_editors/test_stylegan3/test_stylegan3_utils.py

@@ -16,6 +16,13 @@ def test_integer_transformation():
     print(z.shape)
     print(m.shape)
 
+    # cover more lines
+    t = torch.zeros(2)
+    z, m = apply_integer_translation(x, t[0], t[1])
+
+    t = torch.ones(2) * 2
+    z, m = apply_integer_translation(x, t[0], t[1])
+
 
 def test_fractional_translation():
     x = torch.randn(1, 3, 16, 16)
@@ -24,6 +31,13 @@ def test_fractional_translation():
     print(z.shape)
     print(m.shape)
 
+    # cover more lines
+    t = torch.zeros(2)
+    z, m = apply_fractional_translation(x, t[0], t[1])
+
+    t = torch.ones(2) * 2
+    z, m = apply_fractional_translation(x, t[0], t[1])
+
 
 @pytest.mark.skipif(
     digit_version(TORCH_VERSION) < digit_version('1.8.0'),