| 1234567891011121314151617181920212223242526272829303132333435363738394041 |
- # coding=utf-8
- #
- # This file is part of Hypothesis, which may be found at
- # https://github.com/HypothesisWorks/hypothesis-python
- #
- # Most of this work is copyright (C) 2013-2018 David R. MacIver
- # (david@drmaciver.com), but it contains contributions by others. See
- # CONTRIBUTING.rst for a full list of people who may hold copyright, and
- # consult the git log if you need to determine who owns an individual
- # contribution.
- #
- # This Source Code Form is subject to the terms of the Mozilla Public License,
- # v. 2.0. If a copy of the MPL was not distributed with this file, You can
- # obtain one at http://mozilla.org/MPL/2.0/.
- #
- # END HEADER
- from __future__ import division, print_function, absolute_import
- from hypothesis.types import Stream
- from hypothesis.searchstrategy.strategies import SearchStrategy
- class StreamStrategy(SearchStrategy):
- supports_find = False
- def __init__(self, source_strategy):
- super(StreamStrategy, self).__init__()
- self.source_strategy = source_strategy
- def __repr__(self):
- return u'StreamStrategy(%r)' % (self.source_strategy,)
- def do_draw(self, data):
- data.can_reproduce_example_from_repr = False
- def gen():
- while True:
- yield data.draw(self.source_strategy)
- return Stream(gen())
|
