Как смоделировать объект внутри функции main () в python? - PullRequest
Новая
       31

Как смоделировать объект внутри функции main () в python?

0 голосов
/ 04 ноября 2019

Я пытаюсь смоделировать data1 и data2 и пытаюсь предоставить возвращаемое значение. У меня есть следующий код: 

import pandas 

def main():
    data1= pandas.read_excel('path1')
    data2= pandas.read_excel('path2')

if __name__ == '__main__':
    main()

import test1
from unittest.mock import patch
import pandas

class Testdata(unittest.TestCase):

    @patch('test1.main.data1')
    @patch('test1.main.data2')
    def test_main(self, mock_data1, mock_data2):    
        mock_data1.return_value = pandas.DataFrame([some dataframe])
        mock_data2.return_value = pandas.DataFrame([some dataframe])
        test.main()
        data1.assert_called_once()
        data2.assert_called_once()

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

Я получаю следующую ошибку: 

Error
Traceback (most recent call last):
  File "C:\apps\python\3.6.2\lib\unittest\case.py", line 59, in testPartExecutor
    yield
  File "C:\apps\python\3.6.2\lib\unittest\case.py", line 605, in run
    testMethod()
  File "C:\apps\python\3.6.2\lib\unittest\mock.py", line 1171, in patched
    arg = patching.__enter__()
  File "C:\apps\python\3.6.2\lib\unittest\mock.py", line 1227, in __enter__
    self.target = self.getter()
  File "C:\apps\python\3.6.2\lib\unittest\mock.py", line 1397, in <lambda>
    getter = lambda: _importer(target)
  File "C:\apps\python\3.6.2\lib\unittest\mock.py", line 1080, in _importer
    thing = __import__(import_path)
ModuleNotFoundError: No module named 'main'

Как мне решить эту проблему и как смоделировать данные1 и данные2 и предоставитьвернуть ему значение?

1 Ответ

0 голосов
/ 04 ноября 2019

Не могу сказать много, прежде чем смотреть на полный код, но я добавлю import unitest в начале, и это сделает работу.

...