Split out tests into separate file

Author: takluyver

.idea/misc.xml

@@ -286,51 +286,6 @@ def __repr__(self):
             self.__dict__['keydir'],
             ";".join([Path(k).basename() for k in keys]))
 
-
-def test():
-    db = PickleShareDB('~/testpickleshare')
-    db.clear()
-    print("Should be empty:",db.items())
-    db['hello'] = 15
-    db['aku ankka'] = [1,2,313]
-    db['paths/nest/ok/keyname'] = [1,(5,46)]
-    db.hset('hash', 'aku', 12)
-    db.hset('hash', 'ankka', 313)
-    print("12 =",db.hget('hash','aku'))
-    print("313 =",db.hget('hash','ankka'))
-    print("all hashed",db.hdict('hash'))
-    print(db.keys())
-    print(db.keys('paths/nest/ok/k*'))
-    print(dict(db)) # snapsot of whole db
-    db.uncache() # frees memory, causes re-reads later
-
-    # shorthand for accessing deeply nested files
-    lnk = db.getlink('myobjects/test')
-    lnk.foo = 2
-    lnk.bar = lnk.foo + 5
-    print(lnk.bar) # 7
-
-def stress():
-    db = PickleShareDB('~/fsdbtest')
-    import time,sys
-    for i in range(1000):
-        for j in range(1000):
-            if i % 15 == 0 and i < 200:
-                if str(j) in db:
-                    del db[str(j)]
-                continue
-
-            if j%33 == 0:
-                time.sleep(0.02)
-
-            db[str(j)] = db.get(str(j), []) + [(i,j,"proc %d" % os.getpid())]
-            db.hset('hash',j, db.hget('hash',j,15) + 1 )
-
-        print(i, end=' ')
-        sys.stdout.flush()
-        if i % 10 == 0:
-            db.uncache()
-
 def main():
     import textwrap
     usage = textwrap.dedent("""\

.idea/pickleshare.iml

@@ -0,0 +1,54 @@
+from __future__ import print_function
+import os
+
+from pickleshare import PickleShareDB
+
+def test_pickleshare(tmpdir):
+    db = PickleShareDB(tmpdir)
+    db.clear()
+    print("Should be empty:",db.items())
+    assert len(db) == 0
+    db['hello'] = 15
+    assert db['hello'] == 15
+    db['aku ankka'] = [1,2,313]
+    assert db['aku ankka'] == [1,2,313]
+    db['paths/nest/ok/keyname'] = [1,(5,46)]
+    assert db['paths/nest/ok/keyname'] == [1,(5,46)]
+
+    db.hset('hash', 'aku', 12)
+    db.hset('hash', 'ankka', 313)
+    assert db.hget('hash', 'aku') == 12
+    assert db.hget('hash', 'ankka') == 313
+
+    print("all hashed",db.hdict('hash'))
+    print(db.keys())
+    print(db.keys('paths/nest/ok/k*'))
+    print(dict(db)) # snapsot of whole db
+    db.uncache() # frees memory, causes re-reads later
+
+    # shorthand for accessing deeply nested files
+    lnk = db.getlink('myobjects/test')
+    lnk.foo = 2
+    lnk.bar = lnk.foo + 5
+    assert lnk.bar == 7
+
+def test_stress(tmpdir):
+    db = PickleShareDB(tmpdir)
+    import time,sys
+    for i in range(100):
+        for j in range(500):
+            if i % 15 == 0 and i < 70:
+                if str(j) in db:
+                    del db[str(j)]
+                continue
+
+            if j%33 == 0:
+                time.sleep(0.02)
+
+            db[str(j)] = db.get(str(j), []) + [(i,j,"proc %d" % os.getpid())]
+            db.hset('hash',j, db.hget('hash',j,15) + 1 )
+
+        print(i, end=' ')
+        sys.stdout.flush()
+        if i % 10 == 0:
+            db.uncache()