tests(//tests/py): Add a script path test to Python API test suite

Signed-off-by: Naren Dasan <[email protected]>
Signed-off-by: Naren Dasan <[email protected]>

Author: narendasan

tests/py/test_api.py

@@ -34,8 +34,14 @@ def test_compile_traced(self):
         same = (trt_mod(self.input) - self.traced_model(self.input)).abs().max()
         self.assertTrue(same < 2e-3)
 
-    #def test_compile_script(self):
-    #    pass
+    def test_compile_script(self):
+        extra_info = {
+            "input_shapes": [self.input.shape],
+        }
+
+        trt_mod = trtorch.compile(self.scripted_model, extra_info)
+        same = (trt_mod(self.input) - self.scripted_model(self.input)).abs().max()
+        self.assertTrue(same < 2e-3)
 
 class TestCheckMethodOpSupport(unittest.TestCase):
     def setUp(self):