diff --git a/python/pyarrow/tests/test_hdfs.py b/python/pyarrow/tests/test_hdfs.py index 38dd2d54d7f0e..511dbf9a1c4e1 100644 --- a/python/pyarrow/tests/test_hdfs.py +++ b/python/pyarrow/tests/test_hdfs.py @@ -17,7 +17,6 @@ import os import random -import unittest from io import BytesIO from os.path import join as pjoin @@ -80,14 +79,14 @@ def _make_test_file(self, hdfs, test_name, test_path, test_data): return full_path @classmethod - def setUpClass(cls): + def setup_class(cls): cls.check_driver() cls.hdfs = hdfs_test_client() cls.tmp_path = '/tmp/pyarrow-test-{}'.format(random.randint(0, 1000)) cls.hdfs.mkdir(cls.tmp_path) @classmethod - def tearDownClass(cls): + def teardown_class(cls): cls.hdfs.delete(cls.tmp_path, recursive=True) cls.hdfs.close() @@ -391,7 +390,7 @@ def test_write_to_dataset_no_partitions(self): tmpdir, filesystem=self.hdfs) -class TestLibHdfs(HdfsTestCases, unittest.TestCase): +class TestLibHdfs(HdfsTestCases): @classmethod def check_driver(cls):