Skip to content

Commit e476e18

Browse files
committed
Add time_nsorted to benchmark
1 parent a3aed2f commit e476e18

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

asv_bench/benchmarks/frame_methods.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -758,6 +758,9 @@ class NSort:
758758
params = ["first", "last", "all"]
759759
param_names = ["keep"]
760760

761+
def __init__(self):
762+
self.df = None
763+
761764
def setup(self, keep):
762765
self.df = DataFrame(np.random.randn(100000, 3), columns=list("ABC"))
763766

@@ -773,6 +776,12 @@ def time_nsmallest_one_column(self, keep):
773776
def time_nsmallest_two_columns(self, keep):
774777
self.df.nsmallest(100, ["A", "B"], keep=keep)
775778

779+
def time_nsorted_one_column(self, keep):
780+
self.df.nsorted(100, "A", keep=keep, ascending=True)
781+
782+
def time_nsorted_two_columns(self, keep):
783+
self.df.nsorted(100, ["A", "B"], keep=keep, ascending=[True, False])
784+
776785

777786
class Describe:
778787
def setup(self):

0 commit comments

Comments
 (0)