have db tests work again

1 files changed, 5 insertions, 4 deletions

diff --git a/src/db.py b/src/db.py
index 4702def..cabcb80 100755
--- a/src/db.py
+++ b/src/db.py
@@ -117,6 +117,11 @@ class DBIndex(KVStore):
     #             else:
     #                 kv.put([key])
 
+from operator import itemgetter
+def dict_eq(a, b):
+    sort_on_key = lambda d: sorted(d.items(), key=itemgetter(0))
+    return sort_on_key(a) == sort_on_key(b)
+
 if __name__ == '__main__':
     DBDIR = 'test_db.db'
     # TODO: rm -r DBDIR
@@ -165,7 +170,3 @@ if __name__ == '__main__':
         assert(db.get(keys[k]) == data[k])
 
 
-from operator import itemgetter
-def dict_eq(a, b):
-    sort_on_key = lambda d: sorted(d.items(), key=itemgetter(0))
-    return sort_on_key(a) == sort_on_key(b)