def test_range_constructor(self):
bst = pyset(10)
self.assertEqual(bst.to_list(), list(range(10)))
al = pyset(2, 10)
self.assertEqual(al.to_list(), list(range(2, 10)))
ls = pyset(10, 4, -1)
self.assertEqual(ls.to_list(), list(sorted(range(10, 4, -1))))
def test_several_instances(self):
try:
bst = pyset()
nw = pyset()
ny = pyset()
bst.add(4)
nw.add(6.84)
ny.add("Hello world")
except:
self.fail()
def test_from_list(self):
bst = pyset()
ex = list(range(200))
lo = list(range(400, 50000))
ex.extend(lo)
bst.from_list(ex, lo)
self.assertEqual(bst.to_list(), ex)
def test_delete(self):
bst = pyset()
try:
del bst
with self.assertRaises(UnboundLocalError):
bst.add("Hello world")
except:
self.fail()
def test_iterations(self):
expected = []
bst = pyset()
for i in range(1, 100):
bst.add(i)
for i in bst:
expected.append(i)
self.assertEqual(bst.to_list(), expected)
def test_pop(self):
bst = pyset()
with self.assertRaises(IndexError):
bst.pop("Hello world")
bst.add(2)
with self.assertRaises(IndexError):
bst.pop(4)
bst.pop(2)
self.assertEqual(bst.to_list(), [])
def test_add(self):
bst = pyset()
try:
bst.add(3.0)
bst.add(7)
bst.add("lol")
self.assertEqual(bst.size(), 3)
with self.assertRaises(TypeError):
bst.add(3, 40)
except:
self.fail()
#!/usr/bin/env python3
from py_set import pyset
import time
import sys
import random
random_list = [random.randrange(-1000000, 2000000) for _ in range(1000000)]
pst = pyset()
st = set()
lst = list()
print()
print("*** Adding elements ***")
start = time.time()
for i in range(1000000):
lst.append(i)
end = time.time() - start
print("List time on adding 10^6 items: %.2fms" % (end * 1000))
start = time.time()
for i in range(1000000):
st.add(i)
end = time.time() - start
print("Set time on adding 10^6 items: %.2fms" % (end * 1000))
start = time.time()
for i in range(1000000):
pst.add(i)
end = time.time() - start
def test_clear(self):
bst = pyset(4, 100, 10)
bst.clear()
self.assertEqual(bst.to_list(), [])
def test_to_list(self):
bst = pyset()
bst.add(48)
bst.add(38.4)
self.assertEqual(bst.to_list(), [38.4, 48])
def test_size(self):
bst = pyset()
self.assertEqual(bst.size(), 0)
for i in range(10):
bst.add(i)
self.assertEqual(bst.size(), 10)
def test_max_size(self):
bst = pyset()
try:
bst.max_size()
except:
self.fail()
def test_creating_instance(self):
try:
bst = pyset()
except Exception:
self.fail("Can't create an instance of BST")
def test_is_empty(self):
bst = pyset()
self.assertEqual(bst.is_empty(), 1)
bst.add(3)
self.assertEqual(bst.is_empty(), 0)
def test_remove(self):
l = range(1, 100)
bst = pyset(1, 100)
bst.remove(40)
bst.remove("Not in pyset")
self.assertEqual(bst.size(), len(list(l)) - 1)
def test_find(self):
bst = pyset()
bst.add("lol")
self.assertEqual(bst.find("lol"), 1)
self.assertEqual(bst.find(38450), 0)
def test_len(self):
bst = pyset(100)
self.assertEqual(len(bst), 100)
def test_getitem(self):
bst = pyset(10)
self.assertEqual(bst[0], 0)
self.assertEqual(bst[5], 5)
self.assertEqual(bst[9], 9)
def test_contains_method(self):
bst = pyset(100)
res = "NOT IN PYSET" in bst
self.assertEqual(res, 0)
res = 10 in bst
self.assertEqual(res, 1)