Remove a use of checkScript(str) (#35623)

dreiss · facebook-github-bot · commit 7f7fdb101331 · 2020-05-14T10:07:58.000-07:00
Summary: Pull Request resolved: #35623 Python 2 has reached end-of-life and is no longer supported by PyTorch. This test case is valid syntax in Python 3. Test Plan: CI Differential Revision: D20842874 Pulled By: dreiss fbshipit-source-id: 9f12e046f827d4f9d5eca99b0b0b46f73e06ff51
diff --git a/test/test_jit.py b/test/test_jit.py
@@ -13111,15 +13111,12 @@ def subscript_tuple_assign(a, x, i):
 
         self.checkScript(subscript_tuple_assign, ([12, 7, 9, 11], torch.tensor((3, 13, 17)), 0))
 
-        # python 2 does not support star assignments so we use compilation unit to test instead
-        star_code = dedent('''
         def star_tuple_assign():
             # type: () -> Tuple[int, int, Tuple[int, int], Tuple[int, int]]
             a, (b, *c), *d = 1, (2, 3, 4), 5, 6
             return a, b, c, d
-        ''')
 
-        self.checkScript(star_code, (), name='star_tuple_assign')
+        self.checkScript(star_tuple_assign, ())
 
         def subscript_tuple_augmented_assign(a):
             # type: (Tuple[int, int]) -> Tuple[int, int]
