tests(modules): fix tests for pretrained RetNeXt

Author: adosar

src/retnext/modules.py

@@ -55,9 +55,7 @@ class RetNeXt(nn.Module):
     True
 
     >>> # Uses pretrained weights for the backbone.
-    >>> model = RetNeXt(pretrained=True)
-    >>> model.backbone[0].running_mean.item != 0.
-    True
+    >>> model = RetNeXt(pretrained=True)  # doctest: +SKIP
     """
     def __init__(
             self,

tests/test_modules.py

@@ -1,6 +1,16 @@
 import doctest
-import retnext.modules
+import unittest
+
+from retnext import modules
+from retnext.modules import RetNeXt
+
+
+class TestPretrainedRetNeXt(unittest.TestCase):
+    def test_load_pretrained(self):
+        model = RetNeXt(pretrained=True)
+        self.assertNotEqual(model.backbone[0].running_mean.item(), 0)
+
 
 def load_tests(loader, tests, ignore):
-    tests.addTests(doctest.DocTestSuite(retnext.modules))
+    tests.addTests(doctest.DocTestSuite(modules))
     return tests