diff --git a/pydsa/tests/test_binary_search.py b/pydsa/tests/test_binary_search.py index c7b6c59..8da4af9 100644 --- a/pydsa/tests/test_binary_search.py +++ b/pydsa/tests/test_binary_search.py @@ -1,12 +1,25 @@ from pydsa.binary_search import binary_search from random import randint + def test_binary_search(): a = [randint(1, 10) for i in range(10)] item = randint(1, 10) a = sorted(a) ans = binary_search(a, item) - if(ans == -1): + if(ans == -1): assert item not in a - else: + else: assert a[ans] == item + + +def test_binary_search_recursive(): + a = [randint(1, 10) for i in range(10)] + item = randint(1, 10) + a = sorted(a) + ans_through_method_1 = binary_search(a, item, True) + ans_through_method_2 = binary_search(a, item, True, 0, len(a) - 1) + if(ans_through_method_1 == ans_through_method_2 == -1): + assert item not in a + else: + assert a[ans_through_method_1] == item == a[ans_through_method_2] diff --git a/pydsa/tests/test_binary_search_recursive.py b/pydsa/tests/test_binary_search_recursive.py deleted file mode 100644 index ec886e7..0000000 --- a/pydsa/tests/test_binary_search_recursive.py +++ /dev/null @@ -1,13 +0,0 @@ -from pydsa.binary_search import binary_search -from random import randint - -def test_binary_search_recursive(): - a = [randint(1, 10) for i in range(10)] - item = randint(1, 10) - a = sorted(a) - ans1 = binary_search(a, item, True) - ans2 = binary_search(a, item, True, 0, len(a) - 1) - if(ans1 == ans2 == -1): - assert item not in a - else: - assert a[ans1] == item == a[ans2]