From 36a3e92af503765ff64a11cbbcee0a7afc5531a3 Mon Sep 17 00:00:00 2001 From: Carlos Vega <carlos.vega@uni.lu> Date: Tue, 21 Apr 2020 17:41:59 +0200 Subject: [PATCH] add dummy data for virus test --- smash/db_scripts/create_dummy_data.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/smash/db_scripts/create_dummy_data.py b/smash/db_scripts/create_dummy_data.py index 5f9482d8..b6a8ab53 100644 --- a/smash/db_scripts/create_dummy_data.py +++ b/smash/db_scripts/create_dummy_data.py @@ -432,7 +432,18 @@ class smashProvider(BaseProvider): screening_number = self.createSmashScreeningNumber( default_location.prefix) + virus_test_1 = choice([None, True, False], 1, p=[0.2, 0.3, 0.5])[0] + if virus_test_1 is None: + virus_test_1_updated = fake.date_between(start_date='-30d', end_date='-4d') + #inconclusive results have a date + virus_test_1_updated = choice([None, virus_test_1_updated], 1, p=[0.7, 0.3])[0] + elif virus_test_1 == True: + virus_test_1_updated = fake.date_between(start_date='-30d', end_date='-4d') + else: + virus_test_1_updated = fake.date_between(start_date='-30d', end_date='-4d') + study_subject, _ = StudySubject.objects.update_or_create(nd_number=nd_number, subject=subject, + virus_test_1=virus_test_1, virus_test_1_updated=virus_test_1_updated, defaults={'default_location': default_location, 'type': type, 'screening_number': screening_number, 'study': study}) -- GitLab