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)