def test_definition(self): """Tests all ways of defining a FuzzyInteger.""" fuzz = fuzzy.FuzzyInteger(2, 3) for _i in range(20): res = utils.evaluate_declaration(fuzz) self.assertIn(res, [2, 3]) fuzz = fuzzy.FuzzyInteger(4) for _i in range(20): res = utils.evaluate_declaration(fuzz) self.assertIn(res, [0, 1, 2, 3, 4])
class UserProfileFactory(factory.django.DjangoModelFactory): """Factory for UserProfile model.""" date_joined_program = fuzzy.FuzzyDateTime(START_DT) local_name = 'Local Name' birth_date = datetime.date(1992, 1, 1) city = 'User city' region = 'User region' country = fuzzy.FuzzyChoice(COUNTRIES) lat = fuzzy.FuzzyInteger(-90, 90) lon = fuzzy.FuzzyInteger(-180, 180) display_name = factory.Sequence(lambda n: 'UserProfile%s' % n) private_email = factory.Sequence(lambda n: '*****@*****.**' % n) mozillians_profile_url = 'https://mozillians.org/' irc_name = factory.Sequence(lambda n: 'user%s' % n) twitter_account = factory.Sequence(lambda n: 'user%s' % n) facebook_url = factory.Sequence(lambda n: 'http://facebook.com/user%s' % n) linkedin_url = 'http://linkedin.com/profile/' wiki_profile_url = 'https://wiki.mozilla.org/User:'******'userprofile_email_mentor_notification' pre_save.disconnect(email_mentor_notification, UserProfile, dispatch_uid=dispatch_uid) profile = super(UserProfileFactory, cls)._create(target_class, *args, **kwargs) pre_save.connect(email_mentor_notification, UserProfile, dispatch_uid=dispatch_uid) return profile @factory.post_generation def functional_areas(self, create, extracted, **kwargs): """Add functional areas list after object generation.""" if not create: return if extracted: for area in extracted: self.functional_areas.add(area) else: # create random for i in range(randint(1, 6)): area = FunctionalAreaFactory.create() self.functional_areas.add(area) @factory.post_generation def initial_council(self, create, extracted, **kwargs): """Create userprofile with self as mentor.""" if not create: return if extracted: self.mentor = self.user
class AppliedIndicatorFactory(factory.django.DjangoModelFactory): class Meta: model = models.AppliedIndicator indicator = factory.SubFactory(IndicatorBlueprintFactory) lower_result = factory.SubFactory(LowerResultFactory) context_code = fuzzy.FuzzyText(length=5) target = factory.Dict({ 'd': fuzzy.FuzzyInteger(0, 5), 'v': fuzzy.FuzzyInteger(10, 20) })
class SchoolWeeklyStatusFactory(django_factory.DjangoModelFactory): school = SubFactory(SchoolFactory) year = fuzzy.FuzzyInteger(1900, 2200) week = fuzzy.FuzzyInteger(1, 53) connectivity_type = fuzzy.FuzzyText(length=64) connectivity_speed = fuzzy.FuzzyInteger(1, 1000000) connectivity_latency = fuzzy.FuzzyInteger(1, 100) class Meta: model = SchoolWeeklyStatus
def test_definition(self): """Tests all ways of defining a FuzzyInteger.""" fuzz = fuzzy.FuzzyInteger(2, 3) for _i in range(20): res = fuzz.evaluate(2, None, False) self.assertIn(res, [2, 3]) fuzz = fuzzy.FuzzyInteger(4) for _i in range(20): res = fuzz.evaluate(2, None, False) self.assertIn(res, [0, 1, 2, 3, 4])
class CertifiedFilesHistoryFactory(factory.Factory): class Meta: model = jobModels.CertifiedFilesHistory certified_files_history_id = None certify_history_id = fuzzy.FuzzyInteger(9999) submission_id = fuzzy.FuzzyInteger(9999) filename = fuzzy.FuzzyText() file_type_id = fuzzy.FuzzyInteger(9999) warning_filename = fuzzy.FuzzyText() comment = fuzzy.FuzzyText()
class ObjectClassProgramActivityFactory(factory.Factory): class Meta: model = stagingModels.ObjectClassProgramActivity object_class_program_activity_id = None submission_id = fuzzy.FuzzyInteger(9999) job_id = fuzzy.FuzzyInteger(9999) row_number = fuzzy.FuzzyInteger(9999) agency_identifier = fuzzy.FuzzyText() allocation_transfer_agency = fuzzy.FuzzyText() availability_type_code = fuzzy.FuzzyText() beginning_period_of_availa = fuzzy.FuzzyText() by_direct_reimbursable_fun = fuzzy.FuzzyText() deobligations_recov_by_pro_cpe = fuzzy.FuzzyDecimal(9999) ending_period_of_availabil = fuzzy.FuzzyText() gross_outlay_amount_by_pro_cpe = fuzzy.FuzzyDecimal(9999) gross_outlay_amount_by_pro_fyb = fuzzy.FuzzyDecimal(9999) gross_outlays_delivered_or_cpe = fuzzy.FuzzyDecimal(9999) gross_outlays_delivered_or_fyb = fuzzy.FuzzyDecimal(9999) gross_outlays_undelivered_cpe = fuzzy.FuzzyDecimal(9999) gross_outlays_undelivered_fyb = fuzzy.FuzzyDecimal(9999) main_account_code = fuzzy.FuzzyText() object_class = fuzzy.FuzzyText() obligations_delivered_orde_cpe = fuzzy.FuzzyDecimal(9999) obligations_delivered_orde_fyb = fuzzy.FuzzyDecimal(9999) obligations_incurred_by_pr_cpe = fuzzy.FuzzyDecimal(9999) obligations_undelivered_or_cpe = fuzzy.FuzzyDecimal(9999) obligations_undelivered_or_fyb = fuzzy.FuzzyDecimal(9999) program_activity_code = fuzzy.FuzzyText() program_activity_name = fuzzy.FuzzyText() sub_account_code = fuzzy.FuzzyText() ussgl480100_undelivered_or_cpe = fuzzy.FuzzyDecimal(9999) ussgl480100_undelivered_or_fyb = fuzzy.FuzzyDecimal(9999) ussgl480200_undelivered_or_cpe = fuzzy.FuzzyDecimal(9999) ussgl480200_undelivered_or_fyb = fuzzy.FuzzyDecimal(9999) ussgl483100_undelivered_or_cpe = fuzzy.FuzzyDecimal(9999) ussgl483200_undelivered_or_cpe = fuzzy.FuzzyDecimal(9999) ussgl487100_downward_adjus_cpe = fuzzy.FuzzyDecimal(9999) ussgl487200_downward_adjus_cpe = fuzzy.FuzzyDecimal(9999) ussgl488100_upward_adjustm_cpe = fuzzy.FuzzyDecimal(9999) ussgl488200_upward_adjustm_cpe = fuzzy.FuzzyDecimal(9999) ussgl490100_delivered_orde_cpe = fuzzy.FuzzyDecimal(9999) ussgl490100_delivered_orde_fyb = fuzzy.FuzzyDecimal(9999) ussgl490200_delivered_orde_cpe = fuzzy.FuzzyDecimal(9999) ussgl490800_authority_outl_cpe = fuzzy.FuzzyDecimal(9999) ussgl490800_authority_outl_fyb = fuzzy.FuzzyDecimal(9999) ussgl493100_delivered_orde_cpe = fuzzy.FuzzyDecimal(9999) ussgl497100_downward_adjus_cpe = fuzzy.FuzzyDecimal(9999) ussgl497200_downward_adjus_cpe = fuzzy.FuzzyDecimal(9999) ussgl498100_upward_adjustm_cpe = fuzzy.FuzzyDecimal(9999) ussgl498200_upward_adjustm_cpe = fuzzy.FuzzyDecimal(9999) tas = fuzzy.FuzzyText() disaster_emergncy_fund_code = fuzzy.FuzzyText()
class LineScoreFactory(DjangoModelFactory): class Meta: model = LineScore game = factory.SubFactory(GameFactory) game_date_est = factory.LazyAttribute(lambda obj: obj.game.game_date_est) game_sequence = factory.LazyAttribute(lambda obj: obj.game.game_sequence) team = None pts_qtr1 = fuzzy.FuzzyInteger(low=0, high=50) pts_qtr2 = fuzzy.FuzzyInteger(low=0, high=50) pts_qtr3 = fuzzy.FuzzyInteger(low=0, high=50) pts_qtr4 = fuzzy.FuzzyInteger(low=0, high=50) pts_ot1 = FuzzyNullable(non_null_type=fuzzy.FuzzyInteger, prob_null=0.8, low=0, high=50) pts_ot2 = FuzzyNullable(non_null_type=fuzzy.FuzzyInteger, prob_null=0.8, low=0, high=50) pts_ot3 = FuzzyNullable(non_null_type=fuzzy.FuzzyInteger, prob_null=0.8, low=0, high=50) pts_ot4 = FuzzyNullable(non_null_type=fuzzy.FuzzyInteger, prob_null=0.8, low=0, high=50) pts_ot5 = FuzzyNullable(non_null_type=fuzzy.FuzzyInteger, prob_null=0.8, low=0, high=50) pts_ot6 = FuzzyNullable(non_null_type=fuzzy.FuzzyInteger, prob_null=0.8, low=0, high=50) pts_ot7 = FuzzyNullable(non_null_type=fuzzy.FuzzyInteger, prob_null=0.8, low=0, high=50) pts_ot8 = FuzzyNullable(non_null_type=fuzzy.FuzzyInteger, prob_null=0.8, low=0, high=50) pts_ot9 = FuzzyNullable(non_null_type=fuzzy.FuzzyInteger, prob_null=0.8, low=0, high=50) pts_ot10 = FuzzyNullable(non_null_type=fuzzy.FuzzyInteger, prob_null=0.8, low=0, high=50) pts = fuzzy.FuzzyInteger(low=0, high=50)
class CodingResultFactory(factory.django.DjangoModelFactory): class Meta: model = CodingResult coder_facebook_id = "123456" question_solved_id = fuzzy.FuzzyInteger(0, 1000) last_testcase_passed_index = fuzzy.FuzzyInteger(0, 1000) coder_source_code = coding_result_data['coder_source_code'] error = coding_result_data['error'] scores = coding_result_data['scores'] possible_total = fuzzy.FuzzyInteger(0, 1000)
class AssetValueFactory(factory.DjangoModelFactory): class Meta: model = AssetValue # class Params: asset_id = factory.SubFactory(AssetFactory) reporting_date = faker.date() opening_price = fuzzy.FuzzyInteger(0) settlement_price = fuzzy.FuzzyInteger(0) trading_volume = fuzzy.FuzzyInteger(0, high=100) currency = fuzzy.FuzzyText(length=5)
class FABSFactory(factory.Factory): class Meta: model = stagingModels.FABS fabs_id = None afa_generated_unique = fuzzy.FuzzyText() submission_id = fuzzy.FuzzyInteger(9999) job_id = fuzzy.FuzzyInteger(9999) row_number = fuzzy.FuzzyInteger(9999) action_date = fuzzy.FuzzyText() action_type = fuzzy.FuzzyText() assistance_type = fuzzy.FuzzyText() award_description = fuzzy.FuzzyText() awardee_or_recipient_legal = fuzzy.FuzzyText() awardee_or_recipient_uniqu = fuzzy.FuzzyText() awarding_agency_code = fuzzy.FuzzyText() awarding_office_code = fuzzy.FuzzyText() awarding_sub_tier_agency_c = fuzzy.FuzzyText() award_modification_amendme = fuzzy.FuzzyText() business_funds_indicator = fuzzy.FuzzyText() business_types = fuzzy.FuzzyText() correction_delete_indicatr = fuzzy.FuzzyText() face_value_loan_guarantee = fuzzy.FuzzyDecimal(9999) fain = fuzzy.FuzzyText() federal_action_obligation = fuzzy.FuzzyDecimal(9999) funding_agency_code = fuzzy.FuzzyText() funding_office_code = fuzzy.FuzzyText() funding_opportunity_goals = fuzzy.FuzzyText() funding_opportunity_number = fuzzy.FuzzyText() funding_sub_tier_agency_co = fuzzy.FuzzyText() indirect_federal_sharing = fuzzy.FuzzyDecimal(9999) legal_entity_address_line1 = fuzzy.FuzzyText() legal_entity_address_line2 = fuzzy.FuzzyText() legal_entity_country_code = fuzzy.FuzzyText() legal_entity_foreign_city = fuzzy.FuzzyText() legal_entity_foreign_posta = fuzzy.FuzzyText() legal_entity_foreign_provi = fuzzy.FuzzyText() legal_entity_zip5 = fuzzy.FuzzyText() legal_entity_zip_last4 = fuzzy.FuzzyText() non_federal_funding_amount = fuzzy.FuzzyDecimal(9999) original_loan_subsidy_cost = fuzzy.FuzzyDecimal(9999) period_of_performance_curr = fuzzy.FuzzyText() period_of_performance_star = fuzzy.FuzzyText() place_of_performance_code = fuzzy.FuzzyText() place_of_performance_congr = fuzzy.FuzzyText() place_of_perform_country_c = fuzzy.FuzzyText() place_of_performance_forei = fuzzy.FuzzyText() place_of_performance_zip4a = fuzzy.FuzzyText() record_type = fuzzy.FuzzyInteger(1, 2) sai_number = fuzzy.FuzzyText() uri = fuzzy.FuzzyText() is_valid = True
class PlayerFactory(django.DjangoModelFactory): """ Factory-boy for Player Model. """ class Meta: model = PlayerModel pk = fuzzy.FuzzyInteger(min_value, max_value) team = SubFactory(TeamFactory) first_name = fuzzy.FuzzyText(length=10) last_name = fuzzy.FuzzyText(length=10) country = fuzzy.FuzzyText(length=10) jersey_number = fuzzy.FuzzyInteger(min_value, max_value)
class FillLevelFactory(factory.DjangoModelFactory): class Meta: model = EnevoFillLevel time = fuzzy.FuzzyDateTime(start_dt=timezone.now()) fill_level = fuzzy.FuzzyInteger(0, 100) e_site = fuzzy.FuzzyInteger(0, 2000) e_site_name = fuzzy.FuzzyText(length=100) e_site_content_type = fuzzy.FuzzyInteger(0, 5) content_type = fuzzy.FuzzyInteger(0, 5) content_type_name = fuzzy.FuzzyText(length=4) confidence = fuzzy.FuzzyInteger(0, 100)
class InventoryFactory(factory.Factory): class Meta: model = Inventory product_id = factory.Sequence(lambda n: n) condition = fuzzy.FuzzyChoice(choices=keys.CONDITIONS) quantity = fuzzy.FuzzyInteger(keys.QTY_LOW, keys.QTY_HIGH, keys.QTY_STEP) restock_level = fuzzy.FuzzyInteger(keys.QTY_LOW, keys.RESTOCK_LVL, keys.QTY_STEP) available = fuzzy.FuzzyChoice( choices=[keys.AVAILABLE_TRUE, keys.AVAILABLE_FALSE]) ###################################################################### ## Helper ###################################################################### @staticmethod def get_data(type=keys.ATTR_DEFAULT): """ This extracts command line arguments """ if len(sys.argv) < keys.MAX_ATTR + 1: return None, None, None, None, None if type == keys.ATTR_DEFAULT or len(sys.argv) > keys.MAX_ATTR: pid = sys.argv[keys.ATTR_PRODUCT_ID] cnd = sys.argv[keys.ATTR_CONDITION] qty = sys.argv[keys.ATTR_QUANTITY] lvl = sys.argv[keys.ATTR_RESTOCK_LEVEL] avl = sys.argv[keys.ATTR_AVAILABLE] return pid, cnd, qty, lvl, avl sys.argv[type] @staticmethod def read_test_data(): """ Read data for test cases into test_data """ data = [] global test_data_file try: fr = open("./tests/data_to_test.csv", 'r') line = fr.readline() while line: line = fr.readline() row = line.strip().split(',') if row[0]: data.append(row) fr.close() except IOError as err: print(err) finally: return data
class FileRequestFactory(factory.Factory): class Meta: model = jobModels.FileRequest file_request_id = fuzzy.FuzzyInteger(9999) request_date = fuzzy.FuzzyDate(date(2010, 1, 1)) job_id = fuzzy.FuzzyInteger(9999) job = factory.SubFactory(JobFactory) parent_job_id = fuzzy.FuzzyInteger(9999) start_date = fuzzy.FuzzyDate(date(2010, 1, 1)) end_date = fuzzy.FuzzyDate(date(2010, 1, 1)) agency_code = fuzzy.FuzzyText() file_type = fuzzy.FuzzyText() is_cached_file = fuzzy.FuzzyChoice((False, True))
class AlertFactory(factory.DjangoModelFactory): class Meta: model = EnevoAlert id = factory.Sequence(lambda n: n) type = fuzzy.FuzzyInteger(0, 50) type_name = fuzzy.FuzzyText(length=30) reported = fuzzy.FuzzyDateTime(start_dt=timezone.now()) last_observed = fuzzy.FuzzyDateTime(start_dt=timezone.now()) site_name = fuzzy.FuzzyText(length=30) area = fuzzy.FuzzyInteger(0, 50) area_name = fuzzy.FuzzyText(length=30) content_type_name = fuzzy.FuzzyText(length=30) start = fuzzy.FuzzyDateTime(start_dt=timezone.now())
class ContainerTypeFactory(factory.DjangoModelFactory): class Meta: model = EnevoContainerType id = factory.Sequence(lambda n: n) name = fuzzy.FuzzyText(length=40) volume = fuzzy.FuzzyInteger(0, 50) sensor_height = fuzzy.FuzzyFloat(0.5, 100.0) full_height = fuzzy.FuzzyFloat(0.5, 100.0) shape = fuzzy.FuzzyText(length=20) has_bag = fuzzy.FuzzyInteger(0, 1) servicing_amount = fuzzy.FuzzyText(length=20) servicing_method = fuzzy.FuzzyText(length=20) last_modified = fuzzy.FuzzyDateTime(start_dt=timezone.now())
class ItemFactory(factory.Factory): class Meta: model = Item buildyear = fuzzy.FuzzyInteger(0, 9999999999) charges_parkingspace = fuzzy.FuzzyDecimal(0, 99999999999) chargesfinancebasemonth = fuzzy.FuzzyDecimal(0, 99999999999) chargesmaintbasemonth = fuzzy.FuzzyDecimal(0, 99999999999) chargeswater_period = fuzzy.FuzzyText() condition_name = fuzzy.FuzzyChoice([c for c in Condition]) coordinate = factory.List([factory.SubFactory(CoordinateFactory)]) country = fuzzy.FuzzyChoice([c for c in Country]) currency_code = "EUR" cust_itemcode = fuzzy.FuzzyText() debtfreeprice = fuzzy.FuzzyDecimal(0, 99999999999) extralink = factory.List( [factory.SubFactory(ExtraLinkFactory) for _ in range(2)]) floors = fuzzy.FuzzyInteger(0, 9999999999) energyclass = fuzzy.FuzzyText(length=10) holdingtype = fuzzy.FuzzyChoice([c for c in HoldingType]) image = factory.List([factory.SubFactory(ImageFactory) for _ in range(2)]) itemgroup = fuzzy.FuzzyChoice([c for c in ItemGroup]) livingaream2 = fuzzy.FuzzyDecimal(0, 99999999999) loclvlid = 1 locsourceid = 4 lotarea = fuzzy.FuzzyDecimal(0, 99999999999) lotareaunitcode = fuzzy.FuzzyText(length=2) lotholding = fuzzy.FuzzyText() postcode = fuzzy.FuzzyText(length=5) price = fuzzy.FuzzyDecimal(0, 99999999999) price_m2 = fuzzy.FuzzyDecimal(0, 99999999999) quarteroftown = fuzzy.FuzzyText(length=60) realtygroup = fuzzy.FuzzyChoice([c for c in RealtyGroup]) realtyidentifier = fuzzy.FuzzyText(length=40) realty_itemgroup = fuzzy.FuzzyChoice([c for c in ItemGroup]) realtytype = fuzzy.FuzzyChoice([c for c in RealtyType]) realtyoption = factory.List([fuzzy.FuzzyText() for _ in range(2)]) rc_parkingspace_count = fuzzy.FuzzyInteger(0, 9999999999) roomcount = fuzzy.FuzzyInteger(0, 9999999999) scontact = factory.List( [factory.SubFactory(ScontactFactory) for _ in range(2)]) showingdate = fuzzy.FuzzyDateTime(timezone.now()) showing_date2 = fuzzy.FuzzyDateTime(timezone.now()) street = fuzzy.FuzzyText(length=200) supplier_source_itemcode = fuzzy.FuzzyText() text = factory.List([factory.SubFactory(TextFactory) for _ in range(2)]) town = fuzzy.FuzzyText(length=60) tradetype = fuzzy.FuzzyChoice([c for c in TradeType]) zoningname = fuzzy.FuzzyText()
class CardFactory(factory.django.DjangoModelFactory): class Meta: model = models.Card name = factory.Faker('word') image = factory.django.ImageField() attributes = models.Attribute.objects.all().first() rarity = fuzzy.FuzzyInteger(*CARD_RARITY_RANGE) summon_cost = fuzzy.FuzzyInteger(*GENERAL_RANGE) soultrap_cost = fuzzy.FuzzyInteger(*GENERAL_RANGE) magicka_cost = fuzzy.FuzzyInteger(*GENERAL_RANGE) mechanics_text = factory.Faker('word') flavor_text = factory.Faker('sentence') is_collectible = fuzzy.FuzzyChoice([True, False]) expansion = factory.Iterator(models.Expansion.objects.all())
class SubmissionFactory(factory.Factory): class Meta: model = jobModels.Submission submission_id = None user_id = None cgac_code = fuzzy.FuzzyText() reporting_start_date = fuzzy.FuzzyDate(date(2010, 1, 1)) reporting_end_date = fuzzy.FuzzyDate(date(2010, 1, 1)) reporting_fiscal_year = fuzzy.FuzzyInteger(2010, 2040) reporting_fiscal_period = fuzzy.FuzzyInteger(1, 4) is_quarter_format = False publishable = False number_of_errors = 0 number_of_warnings = 0
class ScanFactory(factory.DjangoModelFactory): class Meta: model = models.Scan scan_id = fuzzy.FuzzyInteger(0) latitude = lat1 longitude = lon1 scan_moment = fuzzy.FuzzyDateTime(datetime.datetime(2017, 1, 1, tzinfo=UTC)) parkeervak_id = fuzzy.FuzzyInteger(0, 100) bgt_wegdeel = fuzzy.FuzzyInteger(222, 234) geometrie = Point(uniform(lon1, lon2), uniform(lat1, lat2), srid=4326)
class TransponderFactory(factory.django.DjangoModelFactory): """Transponder model factory.""" description = fuzzy.FuzzyText() alive = fuzzy.FuzzyChoice(choices=[True, False]) uplink_low = fuzzy.FuzzyInteger(200, 500) uplink_high = fuzzy.FuzzyInteger(200, 500) downlink_low = fuzzy.FuzzyInteger(200, 500) downlink_high = fuzzy.FuzzyInteger(200, 500) mode = fuzzy.FuzzyChoice(choices=MODE_CHOICES) invert = fuzzy.FuzzyChoice(choices=[True, False]) baud = fuzzy.FuzzyInteger(4000, 22000, step=1000) satellite = factory.SubFactory(SatelliteFactory) class Meta: model = Transponder
class SitueringFactory(factory.DjangoModelFactory): class Meta(object): model = models.Situering betreft_nummeraanduiding = fuzzy.FuzzyInteger(low=0) # situering_nummeraanduiding = models.CharField(length=128, null=True) eerste_situering = fuzzy.FuzzyAttribute(gen_janee) huisnummer = fuzzy.FuzzyInteger(low=1) if random.randint(0, 50) % 3 == 0: huisletter = fuzzy.FuzzyText(length=1) if random.randint(0, 50) % 3 == 0: huisnummertoevoeging = fuzzy.FuzzyText(length=4) postcode = fuzzy.FuzzyText(length=6) straat = fuzzy.FuzzyText(length=80)
class EventFactory(factory.django.DjangoModelFactory): """Factory for Event model.""" FACTORY_FOR = Event name = factory.Sequence(lambda n: 'Event #%s' % n) start = fuzzy.FuzzyDateTime(START_DT, END_DT) end = fuzzy.FuzzyDateTime(END_DT + datetime.timedelta(days=3)) timezone = fuzzy.FuzzyChoice(common_timezones) venue = 'VenueName' city = 'CityName' region = 'RegionName' country = fuzzy.FuzzyChoice(COUNTRIES) lat = fuzzy.FuzzyInteger(-90, 90) lon = fuzzy.FuzzyInteger(-180, 180) external_link = 'example.com' owner = factory.SubFactory(UserFactory) estimated_attendance = fuzzy.FuzzyChoice(ATTENDANCE_CHOICES) description = 'This is an event description.' extra_content = 'Extra content for event page.' mozilla_event = fuzzy.FuzzyChoice([True, False]) hashtag = 'EventHashtag' converted_visitors = fuzzy.FuzzyInteger(100) swag_bug = factory.SubFactory(BugFactory) budget_bug = factory.SubFactory(BugFactory) times_edited = fuzzy.FuzzyInteger(10) @factory.post_generation def categories(self, create, extracted, **kwargs): """Add event categories after event creation.""" if not create: return if extracted: for category in extracted: self.categories.add(category) else: # add random number of categories for i in range(randint(1, 6)): area = FunctionalAreaFactory.create() self.categories.add(area) @factory.post_generation def metrics(self, create, extracted, **kwargs): """Add event metrics after event creation.""" if not create: return # create 2 metrics by default MetricFactory.create_batch(2, event=self)
class DriveFactory(factory.DjangoModelFactory): class Meta: model = Drive driver = factory.SubFactory(UserFactory) car = factory.SubFactory(CarFactory) date = fuzzy.FuzzyDate((now() - timedelta(days=1000)).date()) start_mileage = fuzzy.FuzzyInteger(1000000) description = factory.Faker('text', max_nb_chars=1000) @factory.lazy_attribute def end_mileage(self): return random.randint(self.start_mileage, 1000000) @factory.post_generation def passengers(self, create, extracted, **kwargs): if not create: return if extracted: for passenger in extracted: self.passengers.add(passenger) @factory.post_generation def project_set(self, create, extracted, **kwargs): if not create: return if extracted: for project in extracted: self.project_set.add(project)
class KiloFactory(factory.DjangoModelFactory): class Meta: model = KilogramWeighMeasurement id = factory.Sequence(lambda n: n) buurt_code = fuzzy.FuzzyText(length=4) stadsdeel = fuzzy.FuzzyText(length=1) seq_id = fuzzy.FuzzyInteger(1, 2000) weigh_at = fuzzy.FuzzyDateTime( datetime.datetime(2018, 10, 16, tzinfo=UTC) ) first_weight = fuzzy.FuzzyInteger(1, 2000) second_weight = fuzzy.FuzzyInteger(1000, 3000) net_weight = fuzzy.FuzzyInteger(1000, 3000) geometrie = get_puntje()
class OfferingFactory(factory.DjangoModelFactory): class Meta: model = models.Offering issue = factory.SubFactory(IssueFactory) unit_price = fuzzy.FuzzyInteger(1, 10) project = factory.SelfAttribute('issue.project') template = factory.SubFactory(OfferingTemplateFactory) @classmethod def get_url(cls, offering=None, action=None): if offering is None: offering = OfferingFactory() url = 'http://testserver' + reverse('support-offering-detail', kwargs={'uuid': offering.uuid.hex}) return url if action is None else url + action + '/' @classmethod def get_list_url(cls): return 'http://testserver' + reverse('support-offering-list') @classmethod def get_list_action_url(cls, action=None): url = 'http://testserver' + reverse('support-offering-list') return url if action is None else url + action + '/'
class DepartmentFactory(factory.DjangoModelFactory): id = factory.Faker('uuid4') name = FuzzyUnicode() number = fuzzy.FuzzyInteger(low=1, high=99) class Meta: model = Department
class DriveFactory(MakeFactoryMixin, DjangoModelFactory): class Meta: model = Drive driver = SubFactory(UserFactory) project = SubFactory(ProjectFactory) passenger = SubFactory(UserFactory) car = SubFactory(CarFactory) date = fuzzy.FuzzyDate((now() - timedelta(days=1000)).date()) start_mileage = fuzzy.FuzzyInteger(1, 1000) end_mileage = fuzzy.FuzzyInteger(1000, 9999) description = Faker("text", max_nb_chars=1000) timestamp = fuzzy.FuzzyInteger(1, 999999999) start_location = Faker("city", locale="uk_UA") end_location = Faker("city", locale="uk_UA") is_verified = False
class RecipeIngredientFactory(django.DjangoModelFactory): recipe = factory.SubFactory(RecipeFactory) ingredient = factory.SubFactory(Ingredient) count = fuzzy.FuzzyInteger(1, 40) class Meta: model = RecipeIngredient