stress test maximum pairwise product in python code example

Example: stress test maximum pairwise product in python

#Uses python3
'trying to work with stress testing'

__author__ = 'InnerPeace'

import random
import argparse

#naive solution
def maxpairwiseproduct(n, a):
    assert len(a) == n, "the length of the list is not equal to n = %r" % n
    assert n >= 2, "'n' should be great or equal than 2"
    result = 0
    for i in range(0, n):
        for j in range(i+1, n):
            if a[i]*a[j] > result:
                result = a[i]*a[j]

    return result

#faster vision of solution
def maxpairwiseproductfast(n, a):
    #assert(len(a) ==  n)
    assert len(a) == n, "the length of the list is not equal to n = %r" % n
    assert n >= 2, "'n' should be great or equal than 2"
    #assert(n >= 2)
    max1 = -1
    max2 = -1
    for i in range(0, n):
        if max1 == -1:
            max1 = i
        elif a[i] >= a[max1]:
            max2 = max1
            max1 = i
        elif max2 == -1 or a[max2] < a[i]:
            max2 = i
    return a[max1] * a[max2]


if __name__ == '__main__':

    parser = argparse.ArgumentParser()
    parser.add_argument("-st", "-stress-test", help="run the stress testing", action="store_true")
    args = parser.parse_args()
    if args.st:
        while True:
            n = random.randint(0, 1000) + 2
            print (n)
            a = []
            for i in range(0, n):
                a.append(random.randint(0, 10000))
            print (a)

            res1 = maxpairwiseproductfast(n, a)
            res2 = maxpairwiseproduct(n, a)

            if res2 != res1:
                print ('wrong answer: %d, %d' % (res1, res2))
                break
            else:
                print ("ok")
    else:
        n = int(input())
        a = [int(x) for x in input().split()]
        print (maxpairwiseproductfast(n, a))
        #print (maxpairwiseproduct(n, a))