test.py

# Test File

from main import *
import itertools
from random import *
from sarr import *
from minheap import *
from maxheap import *
from bst import *

def test():
    s_data.len = 0

    for x in choice([x for x in itertools.permutations(range(100))]):
        s_data.add(x)

    print(str(s_data.contains(3)) + " should be true")

    s_data.remove(3)
    print(str(s_data.contains(3)) + " should be false")

    print(str(s_data.get(0)) + " should be 0")

s_data = SD()

# --------------
# Array Test
# --------------

print "Testing Unsorted Arrays"
s_data.struct = Arr()
test()

# --------------
# Sorted Array Test
# --------------

print "Testing Sorted Arrays\n"

s_data.struct = SArr()
test()

# --------------
# Max Heap Test
# --------------


print "Testing Min Heap\n"

s_data.struct = MinHeap()
test()

# --------------
# Min Heap Test
# --------------


print "Testing Min Heap\n"

s_data.struct = MaxHeap()
test()
# --------------
# Balanced BST Test 
# --------------


print "Testing Balanced BST\n"

s_data.struct = Balanced_BST()
test()