"Тест не пройден: [Errno 2] Нет такого файла или каталога: 'answer.png'" на Gradescope

Question

Я пытаюсь, чтобы пользователь ввел количество синего цвета для океана (от 0,0 до 1,0 с плавающей запятой), а затем попросил его сохранить новый файл изображения в виде png, но Gradescope продолжает возвращать ошибку под названием.

Я попытался изменить ввод на raw_input, но, похоже, это не имеет значения.

blue = input("How blue is the ocean: ")
for row in range(mapShape[0]):
    for col in range(mapShape[1]):
        if elevations[row,col] <= 0:
            #Below sea level
           floodMap[row,col,2] = blue     #Changing 1.0 for amount of blue

output = input("What is the output file: ")
plt.imsave(output, floodMap)

Это сообщения об ошибках из тестов:

Тестирует создание топографической карты со случайным синим цветом> 0,75 - Не удалось

Тестирует создание топографической карты со случайным синим цветом <0,25 - Не удалось </p>

Тестирует создание топографической карты для Нью-Йорка с синим = 0,5 -Сбой

Проверка не пройдена: [Errno 2] Нет такого файла или каталога: 'answer.png'