
from ..statistics import stats_wrapper
from ._testing_data import \
    dataset1, dataset2


def test_wrapper():

    run_wrapper = stats_wrapper(dataset1, dataset2)


import pytest
import numpy as np

from ..statistics import stats_wrapper, statistics_list
from ._testing_data import \
    dataset1, dataset2

spacers = np.arange(2, len(statistics_list) + 1, 2)


# Split these into smaller tests to avoid timeout errors on Travis
@pytest.mark.parametrize(('stats'),
                         [statistics_list[i - 2:i] for i in
                          spacers])
def test_wrapper(stats):

    stats_wrapper(dataset1, dataset2,
                  statistics=stats)

Split wrapper tests into smaller chunks
Split wrapper tests into smaller chunks

  
+ import pytest
+ import numpy as np
+ 
- from ..statistics import stats_wrapper
+ from ..statistics import stats_wrapper, statistics_list
  from ._testing_data import \
      dataset1, dataset2
  
+ spacers = np.arange(2, len(statistics_list) + 1, 2)
  
- def test_wrapper():
  
-     run_wrapper = stats_wrapper(dataset1, dataset2)
+ # Split these into smaller tests to avoid timeout errors on Travis
+ @pytest.mark.parametrize(('stats'),
+                          [statistics_list[i - 2:i] for i in
+                           spacers])
+ def test_wrapper(stats):
  
+     stats_wrapper(dataset1, dataset2,
+                   statistics=stats)
+ 
Split wrapper tests into smaller chunks
## Code Before:

from ..statistics import stats_wrapper
from ._testing_data import \
    dataset1, dataset2


def test_wrapper():

    run_wrapper = stats_wrapper(dataset1, dataset2)

## Instruction:
Split wrapper tests into smaller chunks
## Code After:

import pytest
import numpy as np

from ..statistics import stats_wrapper, statistics_list
from ._testing_data import \
    dataset1, dataset2

spacers = np.arange(2, len(statistics_list) + 1, 2)


# Split these into smaller tests to avoid timeout errors on Travis
@pytest.mark.parametrize(('stats'),
                         [statistics_list[i - 2:i] for i in
                          spacers])
def test_wrapper(stats):

    stats_wrapper(dataset1, dataset2,
                  statistics=stats)
