First commit

Author: Maks3w

.editorconfig

@@ -0,0 +1,24 @@
+# http://editorconfig.org
+
+root = true
+
+[*]
+indent_style = space
+indent_size = 4
+trim_trailing_whitespace = true
+insert_final_newline = true
+charset = utf-8
+end_of_line = lf
+
+[*.py]
+max_line_length = 120
+
+[*.bat]
+indent_style = tab
+end_of_line = crlf
+
+[LICENSE]
+insert_final_newline = false
+
+[Makefile]
+indent_style = tab

.gitignore

@@ -0,0 +1,3 @@
+*.egg-info/
+.tox/
+dist/

.travis.yml

@@ -0,0 +1,28 @@
+sudo: false
+language: python
+
+matrix:
+    include:
+      - python: 3.5
+        env: TOX_ENV=py35
+      - python: 3.6
+        env: TOX_ENV=py36
+      - python: 3.7
+        env: TOX_ENV=py37
+      - python: pypy
+        env: TOX_ENV=pypy
+      - python: 3.6
+        env: TOX_ENV=flake8
+
+install:
+  - pip install tox
+
+script:
+  - tox -e $TOX_ENV
+
+before_cache:
+  - rm -rf $HOME/.cache/pip/log
+
+cache:
+  directories:
+    - $HOME/.cache/pip

MANIFEST.in

@@ -0,0 +1,5 @@
+include LICENSE
+include README.rst
+
+recursive-exclude * __pycache__
+recursive-exclude * *.py[co]

README.rst

@@ -0,0 +1,54 @@
+=============
+pytest-randomness
+=============
+
+`Pytest`_ plugin about random seed management
+
+
+Features
+--------
+
+* TODO
+
+
+Requirements
+------------
+
+* pytest >= 2.8
+
+
+Installation
+------------
+
+You can install "pytest-randomness" via `pip`_ from `PyPI`_::
+
+    $ pip install pytest-randomness
+
+
+Usage
+-----
+
+* TODO
+
+Contributing
+------------
+Contributions are very welcome. Tests can be run with `tox`_, please ensure
+the coverage at least stays the same before you submit a pull request.
+
+License
+-------
+
+Distributed under the terms of the `MIT`_ license, "pytest-randomness" is free and open source software
+
+
+Issues
+------
+
+If you encounter any problems, please `file an issue`_ along with a detailed description.
+
+.. _`MIT`: http://opensource.org/licenses/MIT
+.. _`file an issue`: https://github.com/maks3w/pytest-randomness/issues
+.. _`Pytest`: https://github.com/pytest-dev/pytest
+.. _`tox`: https://tox.readthedocs.io/en/latest/
+.. _`pip`: https://pypi.org/project/pip/
+.. _`PyPI`: https://pypi.org/project

pytest_randomness.py

@@ -0,0 +1,65 @@
+import random
+
+
+class RandomnessPlugin(object):
+    try:
+        from factory import random as factory_random
+        have_factory_boy = True
+    except ImportError:
+        have_factory_boy = False
+
+    try:
+        from faker.generator import random as faker_random
+        have_faker = True
+    except ImportError:
+        have_faker = False
+
+    def __init__(self, config):
+        self.tests = config.cache.get('randomness', {})
+
+    def pytest_runtest_call(self, item):
+        nodeid = item.nodeid
+        if item.config.getoption('randomness_reset_seed') and nodeid in self.tests:
+            self.set_seeds(self.tests[nodeid])
+        else:
+            self.tests[nodeid] = self.get_seeds()
+
+    def pytest_sessionfinish(self, session):
+        session.config.cache.set('randomness', self.tests)
+
+    def get_seeds(self):
+        seeds = {
+            'python': random.getstate(),
+        }
+        if self.have_factory_boy:
+            seeds['factory_boy'] = self.factory_random.get_random_state()
+        if self.have_faker:
+            seeds['faker'] = self.faker_random.getstate()
+        return seeds
+
+    def set_seeds(self, seeds):
+        def seed_to_state(seed):
+            return seed[0], tuple(seed[1]), seed[2]
+
+        if 'python' in seeds:
+            random.setstate(seed_to_state(seeds['python']))
+        if self.have_factory_boy and 'factory_boy' in seeds:
+            self.factory_random.set_random_state(seed_to_state(seeds['factory_boy']))
+        if self.have_faker and 'faker' in seeds:
+            self.faker_random.setstate(seed_to_state(seeds['faker']))
+        return seeds
+
+
+def pytest_configure(config):
+    config.pluginmanager.register(RandomnessPlugin(config), 'randomnessplugin')
+
+
+def pytest_addoption(parser):
+    group = parser.getgroup('random', 'Random seed management')
+
+    group.addoption(
+        '--randomness-reset-seed', action='store_true',
+        dest='randomness_reset_seed', default=False,
+        help="""Do pytest-randomness to reset random.seed() at the
+                start of every individual test."""
+    )

setup.py

@@ -0,0 +1,32 @@
+from setuptools import setup
+
+setup(
+    name='pytest-randomness',
+    version='0.1.0',
+    author='Maks3w',
+    license='MIT',
+    url='https://github.com/Maks3w/pytest-randomness',
+    description='Pytest plugin about random seed management',
+    py_modules=['pytest_randomness'],
+    install_requires=['pytest>=2.8'],
+    classifiers=[
+        'Development Status :: 3 - Alpha',
+        'Framework :: Pytest',
+        'Intended Audience :: Developers',
+        'Topic :: Software Development :: Testing',
+        'Programming Language :: Python :: 3 :: Only',
+        'Programming Language :: Python :: 3',
+        'Programming Language :: Python :: 3.5',
+        'Programming Language :: Python :: 3.6',
+        'Programming Language :: Python :: 3.7',
+        'Programming Language :: Python :: Implementation :: CPython',
+        'Programming Language :: Python :: Implementation :: PyPy',
+        'Operating System :: OS Independent',
+        'License :: OSI Approved :: MIT License',
+    ],
+    entry_points={
+        'pytest11': [
+            'pytest_randomness = pytest_randomness',
+        ],
+    },
+)

tox.ini

@@ -0,0 +1,12 @@
+# For more information about tox, see https://tox.readthedocs.io/en/latest/
+[tox]
+envlist = py27,py34,py35,py36,py37,pypy,flake8
+
+[testenv]
+deps = pytest>=3.0
+commands = pytest {posargs:tests}
+
+[testenv:flake8]
+skip_install = true
+deps = flake8
+commands = flake8 pytest_randomness.py setup.py tests