From 1a919a55c7e2729a0fd3fac3a5766f3e86bf8431 Mon Sep 17 00:00:00 2001
From: Jean-Paul Calderone <exarkun@twistedmatrix.com>
Date: Wed, 29 Apr 2020 20:44:10 -0400
Subject: [PATCH] Remove some more uses of assume()
---
src/_zkapauthorizer/tests/test_controller.py | 12 ++++++------
1 file changed, 6 insertions(+), 6 deletions(-)
diff --git a/src/_zkapauthorizer/tests/test_controller.py b/src/_zkapauthorizer/tests/test_controller.py
index df77bfe..c30f73e 100644
--- a/src/_zkapauthorizer/tests/test_controller.py
+++ b/src/_zkapauthorizer/tests/test_controller.py
@@ -171,14 +171,14 @@ class TokenCountForGroupTests(TestCase):
@given(
redemption_group_counts(),
- integers(min_value=1),
+ integers(min_value=0),
)
- def test_sum(self, num_groups, total_tokens):
+ def test_sum(self, num_groups, extra_tokens):
"""
The sum of the token count for all groups equals the requested total
tokens.
"""
- assume(total_tokens >= num_groups)
+ total_tokens = num_groups + extra_tokens
self.assertThat(
sum(
token_count_for_group(num_groups, total_tokens, group_number)
@@ -190,13 +190,13 @@ class TokenCountForGroupTests(TestCase):
@given(
redemption_group_counts(),
- integers(min_value=1),
+ integers(min_value=0),
)
- def test_well_distributed(self, num_groups, total_tokens):
+ def test_well_distributed(self, num_groups, extra_tokens):
"""
Tokens are distributed roughly evenly across all group numbers.
"""
- assume(total_tokens >= num_groups)
+ total_tokens = num_groups + extra_tokens
lower_bound = total_tokens // num_groups
upper_bound = total_tokens // num_groups + 1
--
GitLab