def test_check_or_generate_pyi__open_function(self):
def mock_open(filename, *args, **kwargs):
if filename == "my_amazing_file.py":
return builtins_io.StringIO("x = 0.0")
else:
return open(filename, *args, **kwargs) # pylint: disable=consider-using-with
options = config.Options.create(
"my_amazing_file.py", check=False, open_function=mock_open)
_, pyi_string, _ = io.check_or_generate_pyi(options)
self.assertEqual(pyi_string, "x: float\n")
def test_check_or_generate_pyi__check(self):
with self._tmpfile("") as f:
options = config.Options.create(f.name, check=True)
_, pyi_string, pytd_ast = io.check_or_generate_pyi(options)
self.assertIsNone(pyi_string)
self.assertIsNone(pytd_ast)
