diff --git a/ClientParamsFactory.py b/ClientParamsFactory.py index bf64057..8ca3388 100644 --- a/ClientParamsFactory.py +++ b/ClientParamsFactory.py @@ -8,11 +8,16 @@ class ClientParamsFactory: def __init__(self) -> None: super().__init__() + def getTrueMore(self, perOfTrue: int) -> bool: + custom = random.randint(0, perOfTrue) + + return custom > 0 + def get_client_params(self) -> ClientParams: payment_delay = random.randint(0, 14) - payed = random.getrandbits(1) + payed = self.getTrueMore(5) net_worth = random.randint(0, 100) - is_skarbowka = random.getrandbits(1) + is_skarbowka = not self.getTrueMore(5) membership = random.getrandbits(1) infuelnce_rate = random.randint(0, 100) is_hat = random.getrandbits(1) diff --git a/main.py b/main.py index b3844e1..21b4e31 100644 --- a/main.py +++ b/main.py @@ -75,7 +75,7 @@ if __name__ == '__main__': writer.writerow(header) - for i in range(50): + for i in range(200): data = test.get_client_params() writer.writerow([data.payment_delay,