find_pairs_test.py

#! /usr/bin/python

from random import randint
from time import time

VALUE_RANGE = (0, 1000)

def timer(f):
    def wrapper(first_list, second_list, required_difference):
        t1 = time()
        result = f(first_list, second_list, required_difference)
        t2 = time()

        n = len(first_list)
        print("executing the function with {} values took ----> {} seconds".format(n, t2 - t1))
        return result
    return wrapper


def test_setup(n):
    first_list = [randint(*VALUE_RANGE) for x in range(n)]
    second_list = [randint(*VALUE_RANGE) for x in range(n)]
    return first_list, second_list

@timer
def test_function(first_list, second_list, required_difference):
    result = 0;

    for x in first_list:
        for y in second_list:
            if (x - y) == required_difference: result+= 1

    return result

def run_test_suite():
    for n in range(1, 100, 10):
        f, s = test_setup(n)
        result = test_function(f, s, 2)
        print("    Number of values found meeting the requirement: {}".format(result))
        print("")

if __name__=="__main__":

    run_test_suite()