
import pkg_resources

import wish_utils


def test_import_modules():
    # normal code path, pytest is a dependency
    distributions = [pkg_resources.get_distribution('pytest')]
    distributions_modules = wish_utils.import_modules(distributions)
    assert len(distributions_modules) == 1
    requirement, modules = distributions_modules[0]
    assert requirement.startswith('pytest==')
    assert set(modules) == {'_pytest', 'pytest'}

    # fail code path, pytest-wish is blacklisted
    distributions = [pkg_resources.get_distribution('pytest-wish')]
    distributions_modules = wish_utils.import_modules(distributions)
    assert len(distributions_modules) == 0


import pkg_resources

import wish_utils


def test_import_coverage():
    """Fix the coverage by pytest-cov, that may trigger after pytest_wish is already imported."""
    from imp import reload  # Python 2 and 3 reload
    import wish_utils
    reload(wish_utils)


def test_import_modules():
    # normal code path, pytest is a dependency
    distributions = [pkg_resources.get_distribution('pytest')]
    distributions_modules = wish_utils.import_modules(distributions)
    assert len(distributions_modules) == 1
    requirement, modules = distributions_modules[0]
    assert requirement.startswith('pytest==')
    assert set(modules) == {'_pytest', 'pytest'}

    # fail code path, pytest-wish is blacklisted
    distributions = [pkg_resources.get_distribution('pytest-wish')]
    distributions_modules = wish_utils.import_modules(distributions)
    assert len(distributions_modules) == 0

Fix pytest-cov coverage of wish_utils.
Fix pytest-cov coverage of wish_utils.

  
  import pkg_resources
  
  import wish_utils
+ 
+ 
+ def test_import_coverage():
+     """Fix the coverage by pytest-cov, that may trigger after pytest_wish is already imported."""
+     from imp import reload  # Python 2 and 3 reload
+     import wish_utils
+     reload(wish_utils)
  
  
  def test_import_modules():
      # normal code path, pytest is a dependency
      distributions = [pkg_resources.get_distribution('pytest')]
      distributions_modules = wish_utils.import_modules(distributions)
      assert len(distributions_modules) == 1
      requirement, modules = distributions_modules[0]
      assert requirement.startswith('pytest==')
      assert set(modules) == {'_pytest', 'pytest'}
  
      # fail code path, pytest-wish is blacklisted
      distributions = [pkg_resources.get_distribution('pytest-wish')]
      distributions_modules = wish_utils.import_modules(distributions)
      assert len(distributions_modules) == 0
  
Fix pytest-cov coverage of wish_utils.
## Code Before:

import pkg_resources

import wish_utils


def test_import_modules():
    # normal code path, pytest is a dependency
    distributions = [pkg_resources.get_distribution('pytest')]
    distributions_modules = wish_utils.import_modules(distributions)
    assert len(distributions_modules) == 1
    requirement, modules = distributions_modules[0]
    assert requirement.startswith('pytest==')
    assert set(modules) == {'_pytest', 'pytest'}

    # fail code path, pytest-wish is blacklisted
    distributions = [pkg_resources.get_distribution('pytest-wish')]
    distributions_modules = wish_utils.import_modules(distributions)
    assert len(distributions_modules) == 0

## Instruction:
Fix pytest-cov coverage of wish_utils.
## Code After:

import pkg_resources

import wish_utils


def test_import_coverage():
    """Fix the coverage by pytest-cov, that may trigger after pytest_wish is already imported."""
    from imp import reload  # Python 2 and 3 reload
    import wish_utils
    reload(wish_utils)


def test_import_modules():
    # normal code path, pytest is a dependency
    distributions = [pkg_resources.get_distribution('pytest')]
    distributions_modules = wish_utils.import_modules(distributions)
    assert len(distributions_modules) == 1
    requirement, modules = distributions_modules[0]
    assert requirement.startswith('pytest==')
    assert set(modules) == {'_pytest', 'pytest'}

    # fail code path, pytest-wish is blacklisted
    distributions = [pkg_resources.get_distribution('pytest-wish')]
    distributions_modules = wish_utils.import_modules(distributions)
    assert len(distributions_modules) == 0
