From ab109e5a691d0163729c2a51fc7674b56ddb17c2 Mon Sep 17 00:00:00 2001 From: Mike Dirolf Date: Thu, 8 Jan 2009 16:29:21 -0500 Subject: [PATCH] gen_mongo_dict returns SON --- test/qcheck.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/test/qcheck.py b/test/qcheck.py index 0c3a4feef..35ea2c2dd 100644 --- a/test/qcheck.py +++ b/test/qcheck.py @@ -6,6 +6,7 @@ import re from objectid import ObjectId from dbref import DBRef +from son import SON gen_target = 100 examples = 5 @@ -13,8 +14,11 @@ examples = 5 def lift(value): return lambda: value -def choose_lifted(list): - return lambda: random.choice(list) +def choose_lifted(generator_list): + return lambda: random.choice(generator_list) + +def map(generator, function): + return lambda: function(generator()) def choose(list): return lambda: random.choice(list)() @@ -111,7 +115,7 @@ def gen_mongo_list(depth): return gen_list(gen_mongo_value(depth - 1), gen_range(0, 10)) def gen_mongo_dict(depth): - return gen_dict(gen_unicode(gen_range(0, 20)), gen_mongo_value(depth - 1), gen_range(0, 10)) + return map(gen_dict(gen_unicode(gen_range(0, 20)), gen_mongo_value(depth - 1), gen_range(0, 10)), SON) def isnt(predicate):