From 81976ce6de5c274dca14159ef9cde8fec4d663a9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Tin=20Tvrtkovi=C4=87?= Date: Tue, 3 Dec 2024 17:25:36 +0100 Subject: [PATCH] test_preconf: always include bools and ints --- tests/test_preconf.py | 12 ++---------- 1 file changed, 2 insertions(+), 10 deletions(-) diff --git a/tests/test_preconf.py b/tests/test_preconf.py index 91610a3f..0197ad0f 100644 --- a/tests/test_preconf.py +++ b/tests/test_preconf.py @@ -198,8 +198,6 @@ def everythings( def native_unions( draw: DrawFn, include_strings=True, - include_bools=True, - include_ints=True, include_floats=True, include_nones=True, include_bytes=True, @@ -207,17 +205,11 @@ def native_unions( include_objectids=False, include_literals=True, ) -> tuple[Any, Any]: - types = [] - strats = {} + types = [bool, int] + strats = {bool: booleans(), int: integers()} if include_strings: types.append(str) strats[str] = text() - if include_bools: - types.append(bool) - strats[bool] = booleans() - if include_ints: - types.append(int) - strats[int] = integers() if include_floats: types.append(float) strats[float] = floats(allow_nan=False)