##// END OF EJS Templates
Add tests for things in utils
Thomas Kluyver -
Show More
@@ -0,0 +1,10 b''
1 from IPython.utils import decorators
2
3 def test_flag_calls():
4 @decorators.flag_calls
5 def f():
6 pass
7
8 assert not f.called
9 f()
10 assert f.called No newline at end of file
@@ -0,0 +1,59 b''
1 import os
2 from unittest import TestCase
3
4 from IPython.testing.decorators import skip
5 from IPython.utils.tempdir import TemporaryDirectory
6 from IPython.utils.pickleshare import PickleShareDB
7
8
9 class PickleShareDBTestCase(TestCase):
10 def setUp(self):
11 self.tempdir = TemporaryDirectory()
12
13 def tearDown(self):
14 self.tempdir.cleanup()
15
16 def test_picklesharedb(self):
17 db = PickleShareDB(self.tempdir.name)
18 db.clear()
19 print("Should be empty:",db.items())
20 db['hello'] = 15
21 db['aku ankka'] = [1,2,313]
22 db['paths/nest/ok/keyname'] = [1,(5,46)]
23 db.hset('hash', 'aku', 12)
24 db.hset('hash', 'ankka', 313)
25 self.assertEqual(db.hget('hash','aku'), 12)
26 self.assertEqual(db.hget('hash','ankka'), 313)
27 print("all hashed",db.hdict('hash'))
28 print(db.keys())
29 print(db.keys('paths/nest/ok/k*'))
30 print(dict(db)) # snapsot of whole db
31 db.uncache() # frees memory, causes re-reads later
32
33 # shorthand for accessing deeply nested files
34 lnk = db.getlink('myobjects/test')
35 lnk.foo = 2
36 lnk.bar = lnk.foo + 5
37 self.assertEqual(lnk.bar, 7)
38
39 @skip("Too slow for regular running.")
40 def test_stress(self):
41 db = PickleShareDB('~/fsdbtest')
42 import time,sys
43 for i in range(1000):
44 for j in range(1000):
45 if i % 15 == 0 and i < 200:
46 if str(j) in db:
47 del db[str(j)]
48 continue
49
50 if j%33 == 0:
51 time.sleep(0.02)
52
53 db[str(j)] = db.get(str(j), []) + [(i,j,"proc %d" % os.getpid())]
54 db.hset('hash',j, db.hget('hash',j,15) + 1 )
55
56 print(i, end=' ')
57 sys.stdout.flush()
58 if i % 10 == 0:
59 db.uncache() No newline at end of file
@@ -280,51 +280,6 b' class PickleShareLink:'
280 280 self.__dict__['keydir'],
281 281 ";".join([Path(k).basename() for k in keys]))
282 282
283
284 def test():
285 db = PickleShareDB('~/testpickleshare')
286 db.clear()
287 print("Should be empty:",db.items())
288 db['hello'] = 15
289 db['aku ankka'] = [1,2,313]
290 db['paths/nest/ok/keyname'] = [1,(5,46)]
291 db.hset('hash', 'aku', 12)
292 db.hset('hash', 'ankka', 313)
293 print("12 =",db.hget('hash','aku'))
294 print("313 =",db.hget('hash','ankka'))
295 print("all hashed",db.hdict('hash'))
296 print(db.keys())
297 print(db.keys('paths/nest/ok/k*'))
298 print(dict(db)) # snapsot of whole db
299 db.uncache() # frees memory, causes re-reads later
300
301 # shorthand for accessing deeply nested files
302 lnk = db.getlink('myobjects/test')
303 lnk.foo = 2
304 lnk.bar = lnk.foo + 5
305 print(lnk.bar) # 7
306
307 def stress():
308 db = PickleShareDB('~/fsdbtest')
309 import time,sys
310 for i in range(1000):
311 for j in range(1000):
312 if i % 15 == 0 and i < 200:
313 if str(j) in db:
314 del db[str(j)]
315 continue
316
317 if j%33 == 0:
318 time.sleep(0.02)
319
320 db[str(j)] = db.get(str(j), []) + [(i,j,"proc %d" % os.getpid())]
321 db.hset('hash',j, db.hget('hash',j,15) + 1 )
322
323 print(i, end=' ')
324 sys.stdout.flush()
325 if i % 10 == 0:
326 db.uncache()
327
328 283 def main():
329 284 import textwrap
330 285 usage = textwrap.dedent("""\
@@ -20,4 +20,20 b' def test_read_file():'
20 20
21 21 read_strip_enc_cookie = openpy.read_py_file(nonascii_path, skip_encoding_cookie=True)
22 22 assert u'coding: iso-8859-5' not in read_strip_enc_cookie
23
23
24 def test_source_to_unicode():
25 with io.open(nonascii_path, 'rb') as f:
26 source_bytes = f.read()
27 nt.assert_equal(openpy.source_to_unicode(source_bytes, skip_encoding_cookie=False),
28 source_bytes.decode('iso-8859-5'))
29
30 source_no_cookie = openpy.source_to_unicode(source_bytes, skip_encoding_cookie=True)
31 nt.assert_not_in(u'coding: iso-8859-5', source_no_cookie)
32
33 def test_list_readline():
34 l = ['a', 'b']
35 readline = openpy._list_readline(l)
36 nt.assert_equal(readline(), 'a')
37 nt.assert_equal(readline(), 'b')
38 with nt.assert_raises(StopIteration):
39 readline() No newline at end of file
@@ -175,3 +175,16 b' def test_strip_email2():'
175 175 src = '> > > list()'
176 176 cln = 'list()'
177 177 nt.assert_equal(text.strip_email_quotes(src), cln)
178
179 def test_LSString():
180 lss = text.LSString("abc\ndef")
181 nt.assert_equal(lss.l, ['abc', 'def'])
182 nt.assert_equal(lss.s, 'abc def')
183
184 def test_SList():
185 sl = text.SList(['a 11', 'b 1', 'a 2'])
186 nt.assert_equal(sl.n, 'a 11\nb 1\na 2')
187 nt.assert_equal(sl.s, 'a 11 b 1 a 2')
188 nt.assert_equal(sl.grep(lambda x: x.startswith('a')), text.SList(['a 11', 'a 2']))
189 nt.assert_equal(sl.fields(0), text.SList(['a', 'b', 'a']))
190 nt.assert_equal(sl.sort(field=1, nums=True), text.SList(['b 1', 'a 2', 'a 11'])) No newline at end of file
General Comments 0
You need to be logged in to leave comments. Login now