import io

import os

try:

    import hypothesis

    import hypothesis.strategies as strategies

except ImportError:

    raise unittest.SkipTest('hypothesis not available')

from .common import (

    make_cffi,

)

s_windowlog = strategies.integers(min_value=zstd.WINDOWLOG_MIN,

                                    max_value=zstd.WINDOWLOG_MAX)

s_chainlog = strategies.integers(min_value=zstd.CHAINLOG_MIN,

                                    max_value=zstd.CHAINLOG_MAX)

s_hashlog = strategies.integers(min_value=zstd.HASHLOG_MIN,

                                max_value=zstd.HASHLOG_MAX)

s_searchlog = strategies.integers(min_value=zstd.SEARCHLOG_MIN,

                                    max_value=zstd.SEARCHLOG_MAX)

s_searchlength = strategies.integers(min_value=zstd.SEARCHLENGTH_MIN,

s_targetlength = strategies.integers(min_value=zstd.TARGETLENGTH_MIN,

s_strategy = strategies.sampled_from((zstd.STRATEGY_FAST,

                                        zstd.STRATEGY_DFAST,

                                        zstd.STRATEGY_GREEDY,

                                        zstd.STRATEGY_LAZY,

                                        zstd.STRATEGY_LAZY2,

                                        zstd.STRATEGY_BTLAZY2,

@make_cffi

@unittest.skipUnless('ZSTD_SLOW_TESTS' in os.environ, 'ZSTD_SLOW_TESTS not set')

class TestCompressionParametersHypothesis(unittest.TestCase):

    @hypothesis.given(s_windowlog, s_chainlog, s_hashlog, s_searchlog,

                        s_searchlength, s_targetlength, s_strategy)

    def test_valid_init(self, windowlog, chainlog, hashlog, searchlog,

                        searchlength, targetlength, strategy):

    @hypothesis.given(s_windowlog, s_chainlog, s_hashlog, s_searchlog,

                        s_searchlength, s_targetlength, s_strategy)

                                                hashlog, searchlog,

                                                searchlength, targetlength,

                                                strategy):

        if searchlength == zstd.SEARCHLENGTH_MIN and strategy in (zstd.STRATEGY_FAST, zstd.STRATEGY_GREEDY):

            searchlength += 1

        elif searchlength == zstd.SEARCHLENGTH_MAX and strategy != zstd.STRATEGY_FAST:

            searchlength -= 1