Add and use an __init__ method for Expenditure objects

[bluechips.git] / bluechips / tests / __init__.py

diff --git a/bluechips/tests/__init__.py b/bluechips/tests/__init__.py
index 514b1a3c9fcef4dbee2b64c4642edac3ec93fed8..cbc0ad459d5eade3873955a405ece11245b539e2 100644 (file)
--- a/bluechips/tests/__init__.py
+++ b/bluechips/tests/__init__.py
@@ -58,9 +58,8 @@ def createExpenditures(n=None):
         n = random.randint(5, 20)
     users = meta.Session.query(bluechips.model.User).all()
     for i in xrange(n):
-        e = bluechips.model.Expenditure()
-        e.spender = random.choice(users)
-        e.amount = Currency(random.randint(1000, 100000))
+        e = bluechips.model.Expenditure(random.choice(users),
+                                        Currency(random.randint(1000, 100000)))
         meta.Session.save(e)
         e.even_split()
     meta.Session.commit()