From e5097f734fd6fa3fbe3df004cec9873760eed0d0 Mon Sep 17 00:00:00 2001 From: blueloveTH Date: Thu, 15 Feb 2024 00:21:51 +0800 Subject: [PATCH] Update 70_random.py --- tests/70_random.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/tests/70_random.py b/tests/70_random.py index 1adafe99..7a4e3e55 100644 --- a/tests/70_random.py +++ b/tests/70_random.py @@ -42,3 +42,17 @@ try: exit(1) except IndexError: pass + +seq = [1, 2, 3, 4] +weights = [0.1, 0.2, 0.2, 0.5] +k = 1000 +res = r.choices(seq, weights, k=k) +assert len(res) == k and isinstance(res, list) + +max_error = 0.03 +for i in range(len(seq)): + actual_w = res.count(seq[i]) / k + assert abs(actual_w - weights[i]) < max_error + + +