diff --git a/test/test_db.py b/test/test_db.py index e00d8fd..06e4780 100644 --- a/test/test_db.py +++ b/test/test_db.py @@ -54,7 +54,7 @@ def test_pointers(db): user = db.save(user) players = db.save(examples.Group(name="players", members=[user])) - user = db.table('User').get(doc_id=user.doc_id) + user = db.table("User").get(doc_id=user.doc_id) assert user.groups.uid == players.uid assert players.members[0].groups.uid == players.uid @@ -73,7 +73,7 @@ def test_subgroups(db): unique_users = set([user for group in starfleet.groups for user in group.members]) - kirk = db.table('User').get(doc_id=kirk.doc_id) + kirk = db.table("User").get(doc_id=kirk.doc_id) assert kirk in unique_users @@ -96,17 +96,14 @@ def test_search(db): # create the crew record crew = db.save(examples.Group(name="Crew", members=[kirk, bones, ricky])) - captain = [m for m in crew.members if m.uid == kirk.uid][0] - assert captain == kirk - User = Query() captains = db.User.search(User.name.matches("Captain")) assert len(captains) == 1 # update the crew members so they have the backreference to crew - kirk = db.table('User').get(doc_id=kirk.doc_id) - bones = db.table('User').get(doc_id=bones.doc_id) - ricky = db.table('User').get(doc_id=ricky.doc_id) + kirk = db.table("User").get(doc_id=kirk.doc_id) + bones = db.table("User").get(doc_id=bones.doc_id) + ricky = db.table("User").get(doc_id=ricky.doc_id) assert kirk in crew.members assert bones in crew.members