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)