def test_get_user(user_row): assert db.insert_user(*user_row) == True user = db.get_user(user_row[0]) assert user is not None assert user == User(*user_row)
def get_user(conn, _id): """SELECT a row from TABLE users, using id""" query = "SELECT * FROM users WHERE id=?" row = conn.execute(query, (_id, )).fetchone() return User(*row) if row else None