def test_not_always_selecting_scope_start(self):
root = source_suite_tree(
'if True:\n a = 1\n if True:\n pass\n'
' else:\n pass\n')
self.assertEqual(3, suites.find_visible_for_suite(root, [4, 6]))
self.assertEqual(3, suites.find_visible_for_suite(root, [3, 5]))
self.assertEqual(3, suites.find_visible_for_suite(root, [4, 5]))
def test_not_always_selecting_scope_start(self):
root = source_suite_tree(
'if True:\n a = 1\n if True:\n pass\n'
' else:\n pass\n')
self.assertEquals(3, suites.find_visible_for_suite(root, [4, 6]))
self.assertEquals(3, suites.find_visible_for_suite(root, [3, 5]))
self.assertEquals(3, suites.find_visible_for_suite(root, [4, 5]))
def test_simple_find_visible_for_different_suites(self):
root = source_suite_tree(
dedent("""\
if True:
pass
a = 1
if False:
pass
"""))
self.assertEqual(1, suites.find_visible_for_suite(root, [2, 3]))
self.assertEqual(5, suites.find_visible_for_suite(root, [5]))
self.assertEqual(1, suites.find_visible_for_suite(root, [2, 5]))
def test_not_always_selecting_scope_start(self):
root = source_suite_tree(
dedent("""\
if True:
a = 1
if True:
pass
else:
pass
"""))
self.assertEqual(3, suites.find_visible_for_suite(root, [4, 6]))
self.assertEqual(3, suites.find_visible_for_suite(root, [3, 5]))
self.assertEqual(3, suites.find_visible_for_suite(root, [4, 5]))
def test_ignoring_classes(self):
root = source_suite_tree(
dedent("""\
a = 1
class C():
pass
"""))
self.assertEqual(1, suites.find_visible_for_suite(root, [1, 3]))
def test_ignoring_functions(self):
root = source_suite_tree(
dedent("""\
def f():
pass
a = 1
"""))
self.assertEqual(3, suites.find_visible_for_suite(root, [2, 3]))
def test_simple_find_visible_for_else(self):
root = source_suite_tree(
dedent("""\
if True:
pass
else: pass
"""))
self.assertEqual(2, suites.find_visible_for_suite(root, [2, 4]))
def test_simple_find_visible_ifs(self):
root = source_suite_tree(
dedent("""\
if True:
a = 1
b = 2
"""))
self.assertEqual(root.find_suite(3), root.find_suite(4))
self.assertEqual(3, suites.find_visible_for_suite(root, [3, 4]))
def test_simple_find_visible_for_different_suites(self):
root = source_suite_tree('if True:\n pass\na = 1\n'
'if False:\n pass\n')
self.assertEqual(1, suites.find_visible_for_suite(root, [2, 3]))
self.assertEqual(5, suites.find_visible_for_suite(root, [5]))
self.assertEqual(1, suites.find_visible_for_suite(root, [2, 5]))
def test_simple_find_visible_for_else(self):
root = source_suite_tree('\nif True:\n pass\nelse: pass\n')
self.assertEqual(2, suites.find_visible_for_suite(root, [2, 4]))
def test_simple_find_visible_ifs(self):
root = source_suite_tree('\nif True:\n a = 1\n b = 2\n')
self.assertEqual(root.find_suite(3), root.find_suite(4))
self.assertEqual(3, suites.find_visible_for_suite(root, [3, 4]))
def test_simple_find_visible_for_else(self):
root = source_suite_tree("\nif True:\n pass\nelse: pass\n")
self.assertEquals(2, suites.find_visible_for_suite(root, [2, 4]))
def test_simple_find_visible_ifs(self):
root = source_suite_tree("\nif True:\n a = 1\n b = 2\n")
self.assertEquals(root.find_suite(3), root.find_suite(4))
self.assertEquals(3, suites.find_visible_for_suite(root, [3, 4]))
def test_ignoring_classes(self):
root = source_suite_tree(
'a = 1\nclass C():\n pass\n')
self.assertEqual(1, suites.find_visible_for_suite(root, [1, 3]))
def test_ignoring_functions(self):
root = source_suite_tree("def f():\n pass\na = 1\n")
self.assertEquals(3, suites.find_visible_for_suite(root, [2, 3]))
def test_simple_find_visible_for_different_suites(self):
root = source_suite_tree("if True:\n pass\na = 1\n" "if False:\n pass\n")
self.assertEquals(1, suites.find_visible_for_suite(root, [2, 3]))
self.assertEquals(5, suites.find_visible_for_suite(root, [5]))
self.assertEquals(1, suites.find_visible_for_suite(root, [2, 5]))
def test_ignoring_functions(self):
root = source_suite_tree(
'def f():\n pass\na = 1\n')
self.assertEqual(3, suites.find_visible_for_suite(root, [2, 3]))
def test_ignoring_classes(self):
root = source_suite_tree("a = 1\nclass C():\n pass\n")
self.assertEquals(1, suites.find_visible_for_suite(root, [1, 3]))
def test_simple_find_visible(self):
root = source_suite_tree('a = 1\n')
self.assertEqual(1, suites.find_visible_for_suite(root, [1]))
def test_simple_find_visible(self):
root = source_suite_tree("a = 1\n")
self.assertEquals(1, suites.find_visible_for_suite(root, [1]))
