diff --git a/project/tests/test_collector.py b/project/tests/test_collector.py index dd35410f..0bf5d8b6 100644 --- a/project/tests/test_collector.py +++ b/project/tests/test_collector.py @@ -1,4 +1,5 @@ import cProfile +import os.path import sys from django.test import TestCase @@ -95,5 +96,5 @@ def _get_prof_file_name(cls, request_path: str) -> str: DataCollector().configure(request) DataCollector().finalise() file_path = DataCollector().request.prof_file.name - filename = file_path.rsplit('/')[-1] + filename = os.path.basename(file_path) return filename.replace(f"{request.id}.prof", "")