Мне нужно закодировать тест для функции, которую я определил в Python 3.9. Поскольку функция должна получать файл определенного расширения и его путь, а затем загружать его в Python, мой тест должен искать любой доступный файл этого расширения и пытаться загрузить его правильно. Как я могу это закодировать?
0
Может быть, это поможет:
import os
file_path = '/' # Location of files
file_extension = 'xlsx'
for i in os.listdir(file_path):
if file_extension == i.split('.')[-1]:
print(i) # The task you want to perform with the specific files
Здесь file_path
является ли путь к списку файлов, file_extension
это конкретное расширение, которое мы ищем.