def test_topn_max_rows(self):
pdf = pd.DataFrame(np.random.rand(2500, 4), columns=['a', 'b', 'c', 'd'])
kdf = koalas.from_pandas(pdf)
data = TopNPlot().get_top_n(kdf)
self.assertEqual(len(data), 2000)
def test_topn_max_rows(self):
pdf = pd.DataFrame(np.random.rand(2500, 4),
columns=["a", "b", "c", "d"])
kdf = ks.from_pandas(pdf)
data = TopNPlot().get_top_n(kdf)
self.assertEqual(len(data), 2000)