From f753377d712c7af64ae840d82923d566b2c7bc8c Mon Sep 17 00:00:00 2001 From: Oleg Broytman Date: Sun, 16 Apr 2017 10:32:23 +0300 Subject: [PATCH] Refactor tests: allow assert_files(one_file_name) --- tests/test_remove_old_files.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/tests/test_remove_old_files.py b/tests/test_remove_old_files.py index 7c0c3c3..e50b6fa 100755 --- a/tests/test_remove_old_files.py +++ b/tests/test_remove_old_files.py @@ -30,11 +30,15 @@ def create_files(files, subdirectory=None): def assert_files_exist(files): + if isinstance(files, str): + files = [files] for fname in files: assert os.path.exists(fname) def assert_files_not_exist(files): + if isinstance(files, str): + files = [files] for fname in files: assert not os.path.exists(fname) @@ -44,8 +48,8 @@ def test_remove_old_files(): assert_files_exist(['test1', 'test2']) os.utime('test2', (0, 0)) assert os.system("remove-old-files.py --older 100 .") == 0 - assert_files_exist(['test1']) - assert_files_not_exist(['test2']) + assert_files_exist('test1') + assert_files_not_exist('test2') def test_recursive(): @@ -55,5 +59,5 @@ def test_recursive(): assert_files_exist([test3, test4]) os.utime(test4, (0, 0)) assert os.system("remove-old-files.py --older 100 .") == 0 - assert_files_exist([test3]) - assert_files_not_exist([test4]) + assert_files_exist(test3) + assert_files_not_exist(test4) -- 2.39.5