From df2bc2a04fb32adb30113402922bd1831d96beb5 Mon Sep 17 00:00:00 2001 From: Jeremi Lisek Date: Thu, 12 May 2022 17:05:12 +0200 Subject: [PATCH] client factory added better data creation --- ClientParamsFactory.py | 9 +++++++-- main.py | 2 +- 2 files changed, 8 insertions(+), 3 deletions(-) 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,