From 5c06d4b5cb8cb495b313364182bcdc46d4f53c9a Mon Sep 17 00:00:00 2001 From: RangiLyu Date: Sat, 17 Jul 2021 17:00:07 +0800 Subject: [PATCH] fix unit test --- tests/test_models/test_fpn/test_fpn.py | 4 ++-- tests/test_models/test_fpn/test_pan.py | 4 ++-- tests/test_models/test_fpn/test_tan.py | 4 ++-- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/test_models/test_fpn/test_fpn.py b/tests/test_models/test_fpn/test_fpn.py index 726c063c5..ddbce1adf 100644 --- 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 index 9286c0d51..fc133d547 100644 --- 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 index db598695c..2281e4a75 100644 --- 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)