Modernize test to use extract_node.

1 files changed, 11 insertions, 18 deletions

diff --git a/astroid/tests/unittest_inference.py b/astroid/tests/unittest_inference.py
index c93b4ee..a520620 100644
--- a/astroid/tests/unittest_inference.py
+++ b/astroid/tests/unittest_inference.py
@@ -664,26 +664,19 @@ class InferenceTest(resources.SysPathSetup, unittest.TestCase):
 
     def test_simple_subscript(self):
         code = '''
-            a = [1, 2, 3][0]
-            b = (1, 2, 3)[1]
-            c = (1, 2, 3)[-1]
-            d = a + b + c
-            print (d)
+            [1, 2, 3][0] #@
+            (1, 2, 3)[1] #@
+            (1, 2, 3)[-1] #@
+            [1, 2, 3][0] + (2, )[0] + (3, )[-1] #@
             e = {'key': 'value'}
-            f = e['key']
-            print (f)
+            e['key'] #@
         '''
-        ast = test_utils.build_module(code, __name__)
-        self.assertEqual([i.value for i in
-                          test_utils.get_name_node(ast, 'a', -1).infer()], [1])
-        self.assertEqual([i.value for i in
-                          test_utils.get_name_node(ast, 'b', -1).infer()], [2])
-        self.assertEqual([i.value for i in
-                          test_utils.get_name_node(ast, 'c', -1).infer()], [3])
-        self.assertEqual([i.value for i in
-                          test_utils.get_name_node(ast, 'd', -1).infer()], [6])
-        self.assertEqual([i.value for i in
-                          test_utils.get_name_node(ast, 'f', -1).infer()], ['value'])
+        ast_nodes = test_utils.extract_node(code, __name__)
+        expected = [1, 2, 3, 6, 'value']
+        for node, expected_value in zip(ast_nodes, expected):
+            inferred = next(node.infer())
+            self.assertIsInstance(inferred, nodes.Const)
+            self.assertEqual(inferred.value, expected_value)
 
     def test_simple_tuple(self):
         module = test_utils.build_module("""