def test_get_label(refresh_db_once, db_connection_sqlalchemy):
df_test = pd.DataFrame()
df_test['label_id'] = [3]
df_test['input_data_id'] = [3]
df_test['in_progress'] = [False]
df_test['user_complete'] = [False]
df_test['needs_improvement'] = [False]
df_test['admin_complete'] = [False]
df_test['paid'] = [False]
df_test['user_comment'] = [None]
df_test['admin_comment'] = [None]
engine = db_connection_sqlalchemy
df = sql_queries.get_label(engine,
user_id=1,
label_task_id=1,
input_data_id=3)
assert_series_equal(df['label_id'], df_test['label_id'])
assert_series_equal(df['input_data_id'], df_test['input_data_id'])
assert_series_equal(df['in_progress'], df_test['in_progress'])
assert_series_equal(df['user_complete'], df_test['user_complete'])
assert_series_equal(df['needs_improvement'], df_test['needs_improvement'])
assert_series_equal(df['admin_complete'], df_test['admin_complete'])
assert_series_equal(df['paid'], df_test['paid'])
assert_series_equal(df['user_comment'], df_test['user_comment'])
assert_series_equal(df['admin_comment'], df_test['admin_comment'])
def test_update_label_status_admin_comment(refresh_db_every_time, db_connection_sqlalchemy):
engine = db_connection_sqlalchemy
label_id = sql_queries.update_label_status(engine, label_id=3, admin_comment='abcdefg')
assert label_id == 3
df = sql_queries.get_label(engine, user_id=1, label_task_id=1, input_data_id=3)
assert df['admin_comment'][0] == 'abcdefg'
def test_update_label_status_do_nothing_if_no_statuses_specified(refresh_db_every_time, db_connection_sqlalchemy):
engine = db_connection_sqlalchemy
label_id = sql_queries.update_label_status(engine, label_id=3)
assert label_id is None
df = sql_queries.get_label(engine, user_id=1, label_task_id=1, input_data_id=3)
assert not df['user_complete'][0]
assert not df['needs_improvement'][0]
assert not df['admin_complete'][0]
assert not df['paid'][0]
def test_update_label_status_admin_complete_and_paid(refresh_db_every_time, db_connection_sqlalchemy):
engine = db_connection_sqlalchemy
label_id = sql_queries.update_label_status(engine, label_id=3, admin_complete=True, paid=True)
assert label_id == 3
df = sql_queries.get_label(engine, user_id=1, label_task_id=1, input_data_id=3)
assert not df['user_complete'][0]
assert not df['needs_improvement'][0]
assert df['admin_complete'][0]
assert df['paid'][0]