AnalyticalSkills/search_student.py

"""
Analytical Skills - opgave zoeken

(c) 2019 Hogeschool Utrecht
Tijmen Muller (tijmen.muller@hu.nl)
"""


import random
from math import floor


def linear_search(lst, target):
    """ Bepaalt of *target* voorkomt in lijst *lst* door middel van lineair zoeken. """
    return False



def linear_search_index(lst, target):
    """ Geeft de positie (m.a.w. index) van *target* in lijst *lst* d.m.v. lineair zoeken. """
    index = 0

    return -1


def linear_search_index_steps(lst, target):
    """ Geeft de positie (m.a.w. index) van *target* in lijst *lst* d.m.v. lineair zoeken, en het aantal benodigde stappen. """
    index = 0
    steps = 0

    return -1, steps


def binary_search(lst, target):
    """ Bepaalt of *target* voorkomt in lijst *lst* door middel van binair zoeken. """
    # stap 1
    mini = 0

    # stap 6
    while False: # hoelang ga je door met zoeken?
        # stap 2
        # stap 3
        return True
        # stap 4
        # stap 5

    return False


def binary_search_index(lst, target):
    """ Geeft de positie (m.a.w. index) van *target* in lijst *lst* d.m.v. binair zoeken. """
    return -1


def binary_search_index_steps(lst, target):
    """ Geeft de positie (m.a.w. index) van *target* in lijst *lst* d.m.v. binair zoeken, en het aantal benodigde stappen. """
    steps = 0
    return (-1, steps)


"""
========================================================================================================================
Onderstaand staan de tests voor je code -- hieronder mag je niets wijzigen!
Je kunt je code testen door deze file te runnen of met behulp van pytest.
"""
def test_linear_search():
    for i in range(10):
        lst_test = random.sample(range(20), 10)
        target = random.randrange(20)
        assert linear_search(lst_test, target) == (target in lst_test), "\x1b[0;31mFout: linear_search({}, {}) geeft {} in plaats van {}".format(lst_test, target, linear_search(lst_test, target), target in lst_test)


def test_linear_search_index():
    for i in range(10):
        lst_test = random.sample(range(20), 10)
        target = random.choice(lst_test)
        assert linear_search_index(lst_test, target) == lst_test.index(target), "\x1b[0;31mFout: linear_search_index({}, {}) geeft {} in plaats van {}".format(lst_test, target, linear_search_index(lst_test, target), lst_test.index(target))

        lst_test = [0, 1, 2]
        assert linear_search_index(lst_test, 3) == -1, "\x1b[0;31mFout: linear_search_index({}, {}) geeft {} in plaats van {}".format(lst_test, 3, linear_search_index(lst_test, 3), -1)


def test_linear_search_index_steps():
    for i in range(10):
        lst_test = random.sample(range(20), 10)
        target = random.choice(lst_test)
        assert linear_search_index_steps(lst_test, target)[0] == lst_test.index(target), "\x1b[0;31mFout: linear_search_index_steps({}, {}) geeft {} in plaats van {}".format(lst_test, target, linear_search_index_steps(lst_test, target)[0], lst_test.index(target))


def test_binary_search():
    for i in range(10):
        lst_test = sorted(random.sample(range(20), 10))
        target = random.randrange(20)
        assert binary_search(lst_test, target) == (target in lst_test), "\x1b[0;31mFout: binary_search({}, {}) geeft {} in plaats van {}".format(lst_test, target, binary_search(lst_test, target), target in lst_test)


def test_binary_search_index():
    for i in range(10):
        lst_test = sorted(random.sample(range(20), 10))
        target = random.choice(lst_test)
        assert binary_search_index(lst_test, target) == lst_test.index(target), "\x1b[0;31mFout: binary_search_index({}, {}) geeft {} in plaats van {}".format(lst_test, target, binary_search_index(lst_test, target), lst_test.index(target))

        lst_test = [0, 1, 2]
        assert binary_search_index(lst_test, 3) == -1, "\x1b[0;31mFout: binary_search_index({}, {}) geeft {} in plaats van {}".format(lst_test, 3, binary_search_index(lst_test, 3), -1)


def test_binary_search_index_steps():
    for i in range(10):
        lst_test = sorted(random.sample(range(20), 10))
        target = random.choice(lst_test)
        assert binary_search_index_steps(lst_test, target)[0] == lst_test.index(target), "\x1b[0;31mFout: binary_search_index_steps({}, {}) geeft {} in plaats van {}".format(lst_test, target, binary_search_index_steps(lst_test, target)[0], lst_test.index(target))


if __name__ == '__main__':
    try:
        test_linear_search()
        print("\x1b[0;32mJe functie linear_search() werkt goed!")

        test_linear_search_index()
        print("\x1b[0;32mJe functie linear_search_index() werkt goed!")

        test_binary_search()
        print("\x1b[0;32mJe functie test_binary_search() werkt goed!")

        test_binary_search_index()
        print("\x1b[0;32mJe functie test_binary_search_index() werkt goed!")

        test_linear_search_index_steps()
        print("\x1b[0;32mJe functie test_linear_search_index_steps() werkt goed!")

        test_binary_search_index_steps()
        print("\x1b[0;32mJe functie test_binary_search_index_steps() werkt goed!")

        size = int(input("\x1b[0;30mGeef een grootte voor de lijst: "))
        lst_test = list(range(size))
        print("Ik ga zoeken in:", lst_test)
        tgt = int(input("Geef een getal: "))

        (idx, cnt) = linear_search_index_steps(lst_test, tgt)
        print("Het lineair zoekalgoritme vond '" + str(tgt) + "' op positie '" + str(idx) + "' na " + str(cnt) + " stappen.")

        (idx, cnt) = binary_search_index_steps(lst_test, tgt)
        print("Het binair zoekalgoritme vond '" + str(tgt) + "' op positie '" + str(idx) + "' na " + str(cnt) + " stappen.")

    except AssertionError as ae:
        print(ae)