Skip to content

Commit c04be15

Browse files
committed
fix test more properly
1 parent 204976e commit c04be15

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

databricks/koalas/tests/test_series.py

+3-4
Original file line numberDiff line numberDiff line change
@@ -255,10 +255,6 @@ def _test_value_counts(self):
255255
self.assertEqual(res.name, exp.name)
256256
self.assert_eq(res, exp, almost=True)
257257

258-
if LooseVersion(pyspark.__version__) < LooseVersion("2.4") and \
259-
default_session().conf.get("spark.sql.execution.arrow.enabled") == "true":
260-
self.assertRaises(RuntimeError, lambda: kser.value_counts())
261-
262258
self.assert_eq(kser.value_counts(normalize=True),
263259
pser.value_counts(normalize=True), almost=True)
264260
self.assert_eq(kser.value_counts(ascending=True),
@@ -405,6 +401,9 @@ def test_value_counts(self):
405401
self._test_value_counts()
406402
finally:
407403
default_session().conf.set("spark.sql.execution.arrow.enabled", "true")
404+
self.assertRaises(
405+
RuntimeError,
406+
lambda: ks.MultiIndex.from_tuples([('x', 'a'), ('x', 'b')]).value_counts())
408407
else:
409408
self._test_value_counts()
410409

0 commit comments

Comments
 (0)