def test_get_virtualenv_directory_is_in_virtualenv(self): with mock.patch('ievv_opensource.utils.virtualenvutils.sys.real_prefix', '/real'): with mock.patch('ievv_opensource.utils.virtualenvutils.sys.prefix', '/virtual'): self.assertEqual('/virtual', virtualenvutils.get_virtualenv_directory())
def test_get_virtualenv_directory_not_in_virtualenv(self): class NoAttributes: pass with mock.patch('ievv_opensource.utils.virtualenvutils.sys', NoAttributes): with self.assertRaises(OSError): virtualenvutils.get_virtualenv_directory()