test Access.referents and Assignment.references

This commit is contained in:
Jimmy Lai 2019-10-04 17:59:48 -07:00 committed by jimmylai
parent 65cc4d23bd
commit 855232e24f

View file

@ -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)