Improve test coverage

Lightning-Universe · Apr 7, 2021 · 41228ba · 41228ba
1 parent e85e8b1
commit 41228ba
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/core/test_model.py b/tests/core/test_model.py
@@ -155,3 +155,8 @@ def test_model_download(tmpdir, cls, filename):
 def test_available_backbones():
     backbones = ImageClassifier.available_backbones()
     assert "resnet152" in backbones
+
+    class Foo(ImageClassifier):
+        backbones = None
+
+    assert Foo.available_backbones() == []