import interleaving as il
import numpy as np
import pytest
np.random.seed(0)
from .test_methods import TestMethods

class TestProbabilisticInterleaveSpeed(TestMethods):

    def test_interleave(self):
        r1 = list(range(100))
        r2 = list(range(100, 200))
        for i in range(1000):
            method = il.Probabilistic([r1, r2])
            ranking = method.interleave()
            print(list(ranking))

import interleaving as il
import numpy as np
import pytest
np.random.seed(0)
from .test_methods import TestMethods

class TestProbabilisticInterleaveSpeed(TestMethods):

    def test_interleave(self):
        r1 = list(range(100))
        r2 = list(range(50, 150))
        r3 = list(range(100, 200))
        r4 = list(range(150, 250))
        for i in range(1000):
            method = il.Probabilistic([r1, r2, r3, r4])
            ranking = method.interleave()
            method.evaluate(ranking, [0, 1, 2])

Add tests for measuring the speed of probabilistic interleaving
Add tests for measuring the speed of probabilistic interleaving

  import interleaving as il
  import numpy as np
  import pytest
  np.random.seed(0)
  from .test_methods import TestMethods
  
  class TestProbabilisticInterleaveSpeed(TestMethods):
  
      def test_interleave(self):
          r1 = list(range(100))
+         r2 = list(range(50, 150))
-         r2 = list(range(100, 200))
+         r3 = list(range(100, 200))
+         r4 = list(range(150, 250))
          for i in range(1000):
-             method = il.Probabilistic([r1, r2])
+             method = il.Probabilistic([r1, r2, r3, r4])
              ranking = method.interleave()
-             print(list(ranking))
+             method.evaluate(ranking, [0, 1, 2])
  
Add tests for measuring the speed of probabilistic interleaving
## Code Before:
import interleaving as il
import numpy as np
import pytest
np.random.seed(0)
from .test_methods import TestMethods

class TestProbabilisticInterleaveSpeed(TestMethods):

    def test_interleave(self):
        r1 = list(range(100))
        r2 = list(range(100, 200))
        for i in range(1000):
            method = il.Probabilistic([r1, r2])
            ranking = method.interleave()
            print(list(ranking))

## Instruction:
Add tests for measuring the speed of probabilistic interleaving
## Code After:
import interleaving as il
import numpy as np
import pytest
np.random.seed(0)
from .test_methods import TestMethods

class TestProbabilisticInterleaveSpeed(TestMethods):

    def test_interleave(self):
        r1 = list(range(100))
        r2 = list(range(50, 150))
        r3 = list(range(100, 200))
        r4 = list(range(150, 250))
        for i in range(1000):
            method = il.Probabilistic([r1, r2, r3, r4])
            ranking = method.interleave()
            method.evaluate(ranking, [0, 1, 2])
