Commit a7d6fa6d authored by F-G Fernandez's avatar F-G Fernandez Committed by Francisco Massa
Browse files

test: Updated assert in test_backbone_utils (#1494)

Updated all raw asserts to corresponding unittest.TestCase.assert. See #1483
parent 0b009ae3
...@@ -15,11 +15,11 @@ class ResnetFPNBackboneTester(unittest.TestCase): ...@@ -15,11 +15,11 @@ class ResnetFPNBackboneTester(unittest.TestCase):
x = torch.rand(1, 3, 300, 300, dtype=self.dtype, device=device) x = torch.rand(1, 3, 300, 300, dtype=self.dtype, device=device)
resnet18_fpn = resnet_fpn_backbone(backbone_name='resnet18', pretrained=False) resnet18_fpn = resnet_fpn_backbone(backbone_name='resnet18', pretrained=False)
y = resnet18_fpn(x) y = resnet18_fpn(x)
assert list(y.keys()) == [0, 1, 2, 3, 'pool'] self.assertEqual(list(y.keys()), [0, 1, 2, 3, 'pool'])
def test_resnet50_fpn_backbone(self): def test_resnet50_fpn_backbone(self):
device = torch.device('cpu') device = torch.device('cpu')
x = torch.rand(1, 3, 300, 300, dtype=self.dtype, device=device) x = torch.rand(1, 3, 300, 300, dtype=self.dtype, device=device)
resnet50_fpn = resnet_fpn_backbone(backbone_name='resnet50', pretrained=False) resnet50_fpn = resnet_fpn_backbone(backbone_name='resnet50', pretrained=False)
y = resnet50_fpn(x) y = resnet50_fpn(x)
assert list(y.keys()) == [0, 1, 2, 3, 'pool'] self.assertEqual(list(y.keys()), [0, 1, 2, 3, 'pool'])
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment