test turned off visual encoder

Author: aaprasad

tests/test_models.py

@@ -50,6 +50,14 @@ def test_encoder_timm():
     input_tensor = torch.rand(b, c, h, w)
     backend = "timm"
 
+    encoder = VisualEncoder(
+        model_name="off", in_chans=c, d_model=features, backend=backend
+    )
+    output = encoder(input_tensor)
+
+    assert output.shape == (b, features)
+    assert not torch.is_nonzero.any()
+
     encoder = VisualEncoder(
         model_name="resnet18", in_chans=c, d_model=features, backend=backend
     )
@@ -93,6 +101,14 @@ def test_encoder_torch():
     input_tensor = torch.rand(b, c, h, w)
     backend = "torch"
 
+    encoder = VisualEncoder(
+        model_name="off", in_chans=c, d_model=features, backend=backend
+    )
+    output = encoder(input_tensor)
+
+    assert output.shape == (b, features)
+    assert not torch.is_nonzero.any()
+
     encoder = VisualEncoder(
         model_name="resnet18", in_chans=c, d_model=features, backend=backend
     )