Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.11/site-packages/hypothesis/strategies/_internal/shared.py: 36%

1# This file is part of Hypothesis, which may be found at 

2# https://github.com/HypothesisWorks/hypothesis/ 

3# 

4# Copyright the Hypothesis Authors. 

5# Individual contributors are listed in AUTHORS.rst and the git log. 

6# 

7# This Source Code Form is subject to the terms of the Mozilla Public License, 

8# v. 2.0. If a copy of the MPL was not distributed with this file, You can 

9# obtain one at https://mozilla.org/MPL/2.0/. 

10 

11from collections.abc import Hashable 

12from typing import Any, Optional 

13 

14from hypothesis.internal.conjecture.data import ConjectureData 

15from hypothesis.strategies._internal import SearchStrategy 

16from hypothesis.strategies._internal.strategies import Ex 

17 

18 

19class SharedStrategy(SearchStrategy[Ex]): 

20 def __init__(self, base: SearchStrategy[Ex], key: Optional[Hashable] = None): 

21 super().__init__() 

22 self.key = key 

23 self.base = base 

24 

25 def __repr__(self) -> str: 

26 if self.key is not None: 

27 return f"shared({self.base!r}, key={self.key!r})" 

28 else: 

29 return f"shared({self.base!r})" 

30 

31 # Ideally would be -> Ex, but key collisions with different-typed values are 

32 # possible. See https://github.com/HypothesisWorks/hypothesis/issues/4301. 

33 def do_draw(self, data: ConjectureData) -> Any: 

34 if self.key is None or getattr(self.base, "_is_singleton", False): 

35 strat_label = id(self.base) 

36 else: 

37 # Assume that uncached strategies are distinguishable by their 

38 # label. False negatives (even collisions w/id above) are ok as 

39 # long as they are infrequent. 

40 strat_label = self.base.label 

41 key = self.key or self 

42 if key not in data._shared_strategy_draws: 

43 drawn = data.draw(self.base) 

44 data._shared_strategy_draws[key] = (strat_label, drawn) 

45 else: 

46 drawn_strat_label, drawn = data._shared_strategy_draws[key] 

47 # Check disabled pending resolution of #4301 

48 if drawn_strat_label != strat_label: # pragma: no cover 

49 pass 

50 # warnings.warn( 

51 # f"Different strategies are shared under {key=}. This" 

52 # " risks drawing values that are not valid examples for the strategy," 

53 # " or that have a narrower range than expected.", 

54 # HypothesisWarning, 

55 # stacklevel=1, 

56 # ) 

57 return drawn