#!/usr/local/bin/python

# Initialize modules
import avl
import xtree
import types
import sys

import timeit

if len(sys.argv) != 3:
	iterations = 1000000
	count = 10
else:
	iterations = int(sys.argv[1])
	count = int(sys.argv[2])

print "Iterations", iterations
print "Repeats", count
print

result_avl = timeit.Timer("""
import avl
a = avl.new()
for i in xrange(%d):
	a.insert(i)
""" % iterations).timeit(count)
print "object avl.new()", result_avl

result_xtree = timeit.Timer("""
import xtree
a = xtree.xtree()
for i in xrange(%d):
	a[i] = True
""" % iterations).timeit(count)
print "object xtree.xtree()", result_xtree

result_dict = timeit.Timer("""
import types
a = dict()
for i in xrange(%d):
	a[i] = True
""" % iterations).timeit(count)
print "object dict()", result_dict

print "      Dict  Xtree AVL"
print "Dict  %.2f  %.2f  %.2f" % (float(result_dict)/result_dict, float(result_dict)/result_xtree, float(result_dict)/result_avl)
print "Xtree %.2f  %.2f  %.2f" % (float(result_xtree)/result_dict, float(result_xtree)/result_xtree, float(result_xtree)/result_avl)
print "AVL   %.2f  %.2f  %.2f" % (float(result_avl)/result_dict, float(result_avl)/result_xtree, float(result_avl)/result_avl)