fix unit test

RangiLyu · Jul 17, 2021 · 5c06d4b · 5c06d4b
1 parent b48e619
commit 5c06d4b
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/tests/test_models/test_fpn/test_fpn.py b/tests/test_models/test_fpn/test_fpn.py
@@ -51,5 +51,5 @@ def test_fpn():
     outs = fpn_model(feats)
     assert len(outs) == fpn_model.num_outs
     for i in range(fpn_model.num_outs):
-        outs[i].shape[1] == out_channels
-        outs[i].shape[2] == outs[i].shape[3] == s // (2 ** i)
+        assert outs[i].shape[1] == out_channels
+        assert outs[i].shape[2] == outs[i].shape[3] == s // (2 ** (i + 1))
diff --git a/tests/test_models/test_fpn/test_pan.py b/tests/test_models/test_fpn/test_pan.py
@@ -51,5 +51,5 @@ def test_pan():
     outs = pan_model(feats)
     assert len(outs) == pan_model.num_outs
     for i in range(pan_model.num_outs):
-        outs[i].shape[1] == out_channels
-        outs[i].shape[2] == outs[i].shape[3] == s // (2 ** i)
+        assert outs[i].shape[1] == out_channels
+        assert outs[i].shape[2] == outs[i].shape[3] == s // (2 ** (i + 1))
diff --git a/tests/test_models/test_fpn/test_tan.py b/tests/test_models/test_fpn/test_tan.py
@@ -40,5 +40,5 @@ def test_tan():
     outs = pan_model(feats)
     assert len(outs) == 3
     for i in range(3):
-        outs[i].shape[1] == out_channels
-        outs[i].shape[2] == outs[i].shape[3] == s // (2 ** i)
+        assert outs[i].shape[1] == out_channels
+        assert outs[i].shape[2] == outs[i].shape[3] == s // (2 ** i)