def test_on_valid_sql():
global df_student
try:
df_student = load_data_from_database(SQL,DB)
except:
df_student = pd.DataFrame()
assert not df_student.empty
import sqlite3
import pandas as pd
sys.path.append("..")
from answers.question_04 import SQL
from supporting_files.data_loader import load_data_from_database
DB = 'supporting_files/hr.db'
ANSWER_SQL = """SELECT first_name,
last_name,
department_id
FROM employees
WHERE department_id IN (3,10)
ORDER BY department_id
"""
df_answer = load_data_from_database(ANSWER_SQL,DB)
def test_on_valid_sql():
global df_student
try:
df_student = load_data_from_database(SQL,DB)
except:
df_student = pd.DataFrame()
assert not df_student.empty
def test_on_number_of_rows():
assert df_student.shape[0] == df_answer.shape[0]
