Python TSTStrの例

コード例 #1

ファイル: benchmark.py プロジェクト: adamhadani/pytrees

def benchmark_insert():
	fh = open("actor_names.txt", "r")
	n=0
	tst = TSTStr()
	
	print "* Inserting actor names DB to tree..."

	start_t = datetime.datetime.now()
	for l in fh:
		tst.insert(l.strip())
		n+=1
	end_t = datetime.datetime.now()

	t = end_t-start_t
	print "* Inserted %i names to tree in %s" % (n, t)

	fh.seek(0)
	d = {}
	n=0
	print "* Inserting actor names DB to hashtable..."

	start_t = datetime.datetime.now()
	for l in fh:
		d[l.strip()] = None
		n+=1
	end_t = datetime.datetime.now()

	t = end_t-start_t

	print "* Inserted %i names to hash dictionary in %s" % (len(d.keys()), t)

	fh.close()

コード例 #2

ファイル: test.py プロジェクト: adamhadani/pytrees

class TestMemory(unittest.TestCase):
	def setUp(self):
		self.tst = TSTStr()
		self.s = "Brad Pitt"
		self.rc_none = sys.getrefcount(None)
		self.rc_s = sys.getrefcount(self.s)

	def testSetup(self):
		self.assertEqual(sys.getrefcount(None), self.rc_none)

	def testInsert(self):
		self.tst.insert(self.s)
		self.assertEqual(self.tst.search(self.s), True)
		self.assertEqual(sys.getrefcount(self.s), self.rc_s)

コード例 #3

ファイル: test.py プロジェクト: adamhadani/pytrees

	def setUp(self):
		self.tst = TSTStr()
		self.s = "Brad Pitt"
		self.rc_none = sys.getrefcount(None)
		self.rc_s = sys.getrefcount(self.s)