def test_john(): conn = psycopg2.connect(database='postgres', host='127.0.0.1', user='******', password='******') cur = conn.cursor() name = query_data(cur, 1) assert name == 'John'
def test_alice(conn): name = query_data(conn, 2) assert name == 'Alice'
def test_john(conn): name = query_data(conn, 1) assert name == 'John'
def test_Mary(): name = query_data(conn, 2) assert name == 'Mary'