diff --git a/smash/db_scripts/create_dummy_data.py b/smash/db_scripts/create_dummy_data.py index 5f9482d80aa2aeaaa34050963f2809dcfe6f1d28..b6a8ab5324685770bd4c06b381cf03df16cdce2b 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})