def test_account_balance_to_date_unordered_set(self):
a1 = app.Account('john')
e1 = app.Transaction(datetime.strptime('2015-01-15', '%Y-%m-%d'), 125,
0, 'mary')
e2 = app.Transaction(datetime.strptime('2015-01-17', '%Y-%m-%d'), 20,
0, 'supermarket')
# unordered by date
a1.add_transactions([e2, e1])
self.assertEqual(a1.transactions[0], e1)
self.assertEqual(a1.transactions[1], e2)
def seed(DBFILENAME):
with sqlite3.connect(DBFILENAME) as conn:
SQL = "DELETE FROM {};"
cur = conn.cursor()
cur.execute(SQL.format('accounts'))
cur.execute(SQL.format('positions'))
cur.execute(SQL.format('trades'))
account = app.Account()
account.username = "******"
account.balance = 1000.0
account.set_password(app.util.hash_pass("password"))
# account.api_key = '0123456789abcde'
account.save()
account = app.Account()
account.username = "******"
account.balance = 900.0
account.set_password(app.util.hash_pass("password"))
account.save()
def seed(DBFILENAME):
with sqlite3.connect(DBFILENAME) as conn:
SQL = "DELETE FROM {};"
cur = conn.cursor()
cur.execute(SQL.format('accounts'))
cur.execute(SQL.format('positions'))
cur.execute(SQL.format('trades'))
account = app.Account()
account.username = "******"
account.balance = 1000.0
account.set_password(app.util.hash_pass("password"))
account.api_key = '0123456789abcde'
account.save()
account = app.Account()
account.username = "******"
account.balance = 900.0
account.set_password(app.util.hash_pass("password"))
account.save()
position = app.Position()
position.account_pk = 1
position.ticker = 'tsla'
position.shares = 5
position.save()
trade1 = app.Trade()
trade1.time = time.time() - 24 * 60 * 60
trade1.ticker = 'tsla'
trade1.account_pk = 1
trade1.volume = 10
trade1.price = app.util.get_price('tsla') - 20.0
trade1.save()
trade2 = app.Trade()
trade2.time = time.time()
trade2.ticker = 'tsla'
trade2.account_pk = 1
trade2.volume = -5
trade2.price = app.util.get_price('tsla') + 20.0
trade2.save()
def test_account_balance_to_date(self):
a1 = app.Account('john')
e1 = app.Transaction(datetime.strptime('2015-01-15', '%Y-%m-%d'), 125,
0, 'mary')
e2 = app.Transaction(datetime.strptime('2015-01-17', '%Y-%m-%d'), 20,
0, 'supermarket')
a1.add_transactions([e1, e2])
self.assertEqual(
a1.balance_to_date(datetime.strptime('2015-01-16', '%Y-%m-%d')),
-125)
self.assertEqual(
a1.balance_to_date(datetime.strptime('2015-01-17', '%Y-%m-%d')),
-145)