From 855232e24f25c1331f735c64fd7ba0130d93a08d Mon Sep 17 00:00:00 2001 From: Jimmy Lai Date: Fri, 4 Oct 2019 17:59:48 -0700 Subject: [PATCH] test Access.referents and Assignment.references --- libcst/metadata/tests/test_scope_provider.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/libcst/metadata/tests/test_scope_provider.py b/libcst/metadata/tests/test_scope_provider.py index 6b5da43b..a2ab5a56 100644 --- a/libcst/metadata/tests/test_scope_provider.py +++ b/libcst/metadata/tests/test_scope_provider.py @@ -802,11 +802,22 @@ class ScopeProviderTest(UnitTest): self.assertEqual( cast(Assignment, list(a_outer_assignments)[0]).node, a_outer_assign ) + self.assertEqual( + {i.node for i in list(a_outer_assignments)[0].accesses}, {a_outer_access} + ) + self.assertEqual( + {i.node for i in list(a_outer_assignments)[0].references}, {a_outer_access} + ) a_outer_assesses = scope_of_module.accesses[a_outer_assign] self.assertEqual(len(a_outer_assesses), 1) self.assertEqual(list(a_outer_assesses)[0].node, a_outer_access) + self.assertEqual( + {cast(Assignment, i).node for i in list(a_outer_assesses)[0].referents}, + {a_outer_assign}, + ) + self.assertTrue(a_outer_assign in scope_of_module.accesses) self.assertTrue(a_outer_assign in scope_of_module.assignments) self.assertTrue(a_outer_access in scope_of_module.accesses) @@ -860,3 +871,5 @@ class ScopeProviderTest(UnitTest): {cast(Assignment, i).node for i in scope_of_g.assignments}, {b_inner_most_assign}, ) + + self.assertEqual(len(set(scopes.values())), 3)