diff --git a/tests/algo/base.py b/tests/algo/base.py index 94cf1a2..b49ad8f 100644 --- a/tests/algo/base.py +++ b/tests/algo/base.py @@ -65,16 +65,8 @@ def train(self): class TestBase(unittest.TestCase): @classmethod def setUpClass(cls): - if not os.path.isdir('ext/ml-100k/'): - raise RuntimeError('Cannot find the ./ext/ml-100k directory') cls.ml_100k = './ext/ml-100k/' - - if not os.path.isdir('ext/ml-20m'): - raise RuntimeError('Cannot find the ./ext/ml-20m directory') cls.ml_20m = './ext/ml-20m/' - - if not os.path.isdir('ext/text8'): - raise RuntimeError('Cannot find the ./ext/text8 directory') cls.text8 = './ext/text8/' cls.temp_files = []