Ejemplos de Mock.__file__ en Python

Ejemplo n.º 1

Archivo: test_imports.py Proyecto: ChillarAnand/celery

def test_module_file():
    m1 = Mock()
    m1.__file__ = '/opt/foo/xyz.pyc'
    assert module_file(m1) == '/opt/foo/xyz.py'
    m2 = Mock()
    m2.__file__ = '/opt/foo/xyz.py'
    assert module_file(m1) == '/opt/foo/xyz.py'

Ejemplo n.º 2

Archivo: test_celery.py Proyecto: Scalr/celery

 def test_prepare_prog_name(self):
     x = CeleryCommand(app=self.app)
     main = Mock(name='__main__')
     main.__file__ = '/opt/foo.py'
     with patch.dict(sys.modules, __main__=main):
         assert x.prepare_prog_name('__main__.py') == '/opt/foo.py'
         assert x.prepare_prog_name('celery') == 'celery'