def test_model_task_redis_no_dupes_data_unassign_assigned_data(
test_project_labeled_and_tfidf, test_queue_labeled,
test_irr_queue_labeled, test_admin_queue_labeled, test_redis, tmpdir,
settings):
project = test_project_labeled_and_tfidf
person2 = create_profile('test_profilezzz', 'password',
'*****@*****.**')
person3 = create_profile('test_profile2', 'password',
'*****@*****.**')
ProjectPermissions.objects.create(profile=person2,
project=project,
permission='CODER')
ProjectPermissions.objects.create(profile=person3,
project=project,
permission='CODER')
initial_training_set = project.get_current_training_set().set_number
queue = project.queue_set.get(type="normal")
queue.length = 40
queue.save()
irr_queue = project.queue_set.get(type="irr")
irr_queue.length = 40
irr_queue.save()
model_path_temp = tmpdir.listdir()[0].mkdir('model_pickles')
settings.MODEL_PICKLE_PATH = str(model_path_temp)
batch_size = project.batch_size
fill_queue(queue,
'random',
irr_queue,
irr_percent=project.percentage_irr,
batch_size=batch_size)
labels = project.labels.all()
assignments = get_assignments(project.creator, project, batch_size)
for assignment in assignments:
label_data(random.choice(labels), assignment, project.creator, 3)
tasks.send_model_task.delay(project.pk).get()
assert project.get_current_training_set(
).set_number == initial_training_set + 1
redis_items = test_redis.lrange(redis_serialize_queue(queue), 0, -1)
assert len(redis_items) == len(set(redis_items))
assignments = get_assignments(project.creator, project, 40)
for assignment in assignments[:batch_size]:
label_data(random.choice(labels), assignment, project.creator, 3)
tasks.send_model_task.delay(project.pk).get()
assert project.get_current_training_set(
).set_number == initial_training_set + 2
redis_items = test_redis.lrange(redis_serialize_queue(queue), 0, -1)
assert len(redis_items) == len(set(redis_items))
batch_unassign(project.creator)
redis_items = test_redis.lrange(redis_serialize_queue(queue), 0, -1)
assert len(redis_items) == len(set(redis_items))
def test_create_profile(db):
username = '******'
password = '******'
email = '*****@*****.**'
create_profile(username, password, email)
auth_user_attrs = {
'username': username,
'password': password,
'email': email
}
assert_obj_exists(get_user_model(), auth_user_attrs)
auth_user = (get_user_model().objects.filter(**auth_user_attrs).first())
assert_obj_exists(Profile, {'user': auth_user})
def test_create_profile(db):
username = "******"
password = "******"
email = "*****@*****.**"
create_profile(username, password, email)
auth_user_attrs = {
"username": username,
"password": password,
"email": email
}
assert_obj_exists(get_user_model(), auth_user_attrs)
auth_user = get_user_model().objects.filter(**auth_user_attrs).first()
assert_obj_exists(Profile, {"user": auth_user})
def test_profile3(db):
"""Additional profile for tests requiring multiple users."""
return create_profile("test_profile3", "password", "*****@*****.**")
def test_profile(db):
"""Creates a test profile with associated auth_user."""
return create_profile("test_profile", "password", "*****@*****.**")
def test_profile2(db):
'''
Additional profile for tests requiring multiple users.
'''
return create_profile('test_profile2', 'password', '*****@*****.**')
def test_profile(db):
'''
Creates a test profile with associated auth_user.
'''
return create_profile('test_profile', 'password', '*****@*****.**')