Skip to content

Commit d36e2ba

Browse files
committed
use create_random_string()
1 parent d45ba7a commit d36e2ba

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

test/test_datasets.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -962,10 +962,13 @@ def inject_fake_data(self, tmpdir, config):
962962
classes = ("Abseiling", "Zumba")
963963
num_videos_per_class = 2
964964

965-
letters = string.ascii_letters + string.digits + "-_"
965+
digits = string.ascii_letters + string.digits + "-_"
966966
for cls in classes:
967967
datasets_utils.create_video_folder(
968-
tmpdir, cls, lambda _: f"{''.join(random.choice(letters) for _ in range(11))}.avi", num_videos_per_class
968+
tmpdir,
969+
cls,
970+
lambda _: f"{datasets_utils.create_random_string(11, digits)}.avi",
971+
num_videos_per_class,
969972
)
970973

971974
return num_videos_per_class * len(classes)

0 commit comments

Comments
 (0)