mirror of
https://github.com/rdkit/rdkit.git
synced 2026-06-04 21:54:27 +08:00
68 lines
1.7 KiB
Python
Executable File
68 lines
1.7 KiB
Python
Executable File
# $Id: UnitTestTopNContainer.py 5009 2006-02-22 15:19:49Z glandrum $
|
|
#
|
|
# Copyright (C) 2003-2006 Rational Discovery LLC
|
|
#
|
|
# @@ All Rights Reserved @@
|
|
#
|
|
import unittest
|
|
import random
|
|
from DataStructs.TopNContainer import TopNContainer
|
|
|
|
class TestCase(unittest.TestCase):
|
|
def test1(self):
|
|
""" simple test with a known answer """
|
|
cont = TopNContainer(4)
|
|
for foo in range(10):
|
|
cont.Insert(foo,str(foo))
|
|
assert cont.GetPts()==range(6,10)
|
|
assert cont.GetExtras()==[str(x) for x in range(6,10)]
|
|
def test2(self):
|
|
""" larger scale random test """
|
|
cont = TopNContainer(50)
|
|
for i in range(1000):
|
|
cont.Insert(random.random())
|
|
vs = cont.GetPts()
|
|
last = vs.pop(0)
|
|
while vs:
|
|
assert vs[0]>=last
|
|
last = vs.pop(0)
|
|
def test3(self):
|
|
""" random test with extras"""
|
|
cont = TopNContainer(10)
|
|
for i in range(100):
|
|
v = random.random()
|
|
cont.Insert(v,v+1)
|
|
vs = cont.GetExtras()
|
|
last = vs.pop(0)
|
|
while vs:
|
|
assert vs[0]>=last
|
|
last = vs.pop(0)
|
|
def test4(self):
|
|
""" random test with extras and getitem"""
|
|
cont = TopNContainer(10)
|
|
for i in range(100):
|
|
v = random.random()
|
|
cont.Insert(v,v+1)
|
|
lastV,lastE = cont[0]
|
|
for i in range(1,len(cont)):
|
|
v,e = cont[i]
|
|
assert v>=lastV
|
|
assert e>=lastE
|
|
lastV,lastE = v,e
|
|
def test5(self):
|
|
""" random test with extras and getitem, include reverse"""
|
|
cont = TopNContainer(10)
|
|
for i in range(100):
|
|
v = random.random()
|
|
cont.Insert(v,v+1)
|
|
cont.reverse()
|
|
lastV,lastE = cont[0]
|
|
for i in range(1,len(cont)):
|
|
v,e = cont[i]
|
|
assert v<=lastV
|
|
assert e<=lastE
|
|
lastV,lastE = v,e
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|