diff options
author | Claudiu Popa <cpopa@cloudbasesolutions.com> | 2015-07-01 19:10:28 +0300 |
---|---|---|
committer | Claudiu Popa <cpopa@cloudbasesolutions.com> | 2015-07-01 19:10:28 +0300 |
commit | 2eed56e96aaa90e9ce90404afd4833a527f44dd7 (patch) | |
tree | 6e7f79d2262a47d9dfa9d8e583a7a43b98373d78 | |
parent | 6982660978595a4a9b5fb616e6a683100be8e51b (diff) | |
download | astroid-2eed56e96aaa90e9ce90404afd4833a527f44dd7.tar.gz |
Modernize test to use extract_node.
-rw-r--r-- | astroid/tests/unittest_inference.py | 29 |
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(""" |