@@ -627,49 +627,63 @@ def time_first(self):
627
627
628
628
629
629
class TransformEngine :
630
- def setup (self ):
630
+
631
+ param_names = ["parallel" ]
632
+ params = [[True , False ]]
633
+
634
+ def setup (self , parallel ):
631
635
N = 10 ** 3
632
636
data = DataFrame (
633
637
{0 : [str (i ) for i in range (100 )] * N , 1 : list (range (100 )) * N },
634
638
columns = [0 , 1 ],
635
639
)
640
+ self .parallel = parallel
636
641
self .grouper = data .groupby (0 )
637
642
638
- def time_series_numba (self ):
643
+ def time_series_numba (self , parallel ):
639
644
def function (values , index ):
640
645
return values * 5
641
646
642
- self .grouper [1 ].transform (function , engine = "numba" )
647
+ self .grouper [1 ].transform (
648
+ function , engine = "numba" , engine_kwargs = {"parallel" : self .parallel }
649
+ )
643
650
644
- def time_series_cython (self ):
651
+ def time_series_cython (self , parallel ):
645
652
def function (values ):
646
653
return values * 5
647
654
648
655
self .grouper [1 ].transform (function , engine = "cython" )
649
656
650
- def time_dataframe_numba (self ):
657
+ def time_dataframe_numba (self , parallel ):
651
658
def function (values , index ):
652
659
return values * 5
653
660
654
- self .grouper .transform (function , engine = "numba" )
661
+ self .grouper .transform (
662
+ function , engine = "numba" , engine_kwargs = {"parallel" : self .parallel }
663
+ )
655
664
656
- def time_dataframe_cython (self ):
665
+ def time_dataframe_cython (self , parallel ):
657
666
def function (values ):
658
667
return values * 5
659
668
660
669
self .grouper .transform (function , engine = "cython" )
661
670
662
671
663
672
class AggEngine :
664
- def setup (self ):
673
+
674
+ param_names = ["parallel" ]
675
+ params = [[True , False ]]
676
+
677
+ def setup (self , parallel ):
665
678
N = 10 ** 3
666
679
data = DataFrame (
667
680
{0 : [str (i ) for i in range (100 )] * N , 1 : list (range (100 )) * N },
668
681
columns = [0 , 1 ],
669
682
)
683
+ self .parallel = parallel
670
684
self .grouper = data .groupby (0 )
671
685
672
- def time_series_numba (self ):
686
+ def time_series_numba (self , parallel ):
673
687
def function (values , index ):
674
688
total = 0
675
689
for i , value in enumerate (values ):
@@ -679,9 +693,11 @@ def function(values, index):
679
693
total += value * 2
680
694
return total
681
695
682
- self .grouper [1 ].agg (function , engine = "numba" )
696
+ self .grouper [1 ].agg (
697
+ function , engine = "numba" , engine_kwargs = {"parallel" : self .parallel }
698
+ )
683
699
684
- def time_series_cython (self ):
700
+ def time_series_cython (self , parallel ):
685
701
def function (values ):
686
702
total = 0
687
703
for i , value in enumerate (values ):
@@ -693,7 +709,7 @@ def function(values):
693
709
694
710
self .grouper [1 ].agg (function , engine = "cython" )
695
711
696
- def time_dataframe_numba (self ):
712
+ def time_dataframe_numba (self , parallel ):
697
713
def function (values , index ):
698
714
total = 0
699
715
for i , value in enumerate (values ):
@@ -703,9 +719,11 @@ def function(values, index):
703
719
total += value * 2
704
720
return total
705
721
706
- self .grouper .agg (function , engine = "numba" )
722
+ self .grouper .agg (
723
+ function , engine = "numba" , engine_kwargs = {"parallel" : self .parallel }
724
+ )
707
725
708
- def time_dataframe_cython (self ):
726
+ def time_dataframe_cython (self , parallel ):
709
727
def function (values ):
710
728
total = 0
711
729
for i , value in enumerate (values ):
0 commit comments