У объекта 'module' нет атрибута '41531' Ошибка блока данных Python Notebook

Question

Я пытаюсь запустить простой модульный тест в блокноте Python в блоке данных Azure.

import unittest

class KnownValues(unittest.TestCase):
known_values =  ((1, 'I'),
                 (2, 'II'),
                 (3, 'III'),
                 (4, 'IV'),
                 (5, 'V'))

def test_to_roman_values(self):
    for integer, numeral in self.known_values:
        result = roman1.to_roman(integer)
        self.assertEqual(numeral, result)

if __name__ == '__main__':
    unittest.main()

но это выдает мне ошибку в строке unittest.main () - это "модуль" объект не имеет атрибута "41531" В чем проблема?