geoffkoh

@pytest.fixture
def setup_database():
    """ Fixture to set up the in-memory database with test data """
    conn = sqlite3.connect(':memory:')
    cursor = conn.cursor()
    cursor.execute('''
        CREATE TABLE stocks
        (date text, trans text, symbol text, qty real, price real)''')
    yield conn

geoffkoh

# Third party imports
from sqlalchemy import Column, Integer, String, DateTime, Float, ForeignKey
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import relationship
from sqlalchemy.sql import func

Base = declarative_base()


class User(Base):

geoffkoh

# Standard imports
from typing import List

# Third party imports
from sqlalchemy.orm.session import Session
from sqlalchemy.orm.exc import NoResultFound

# Application imports
from models import Account, User, Transaction

geoffkoh

# Third party imports
import pytest
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

# Application imports
from core import get_user, get_accounts_by_user, compute_balance, debit
from models import Base, User, Account, UserAccount, Transaction

geoffkoh

# Standard imports 
import tempfile

# Third party imports
import pytest
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from pytest_postgresql import factories