def test_is_entry_point_false() -> None: assert not is_entry_point( textwrap.dedent( """ # Note commented out. # if __name__ == "__main__": # main() """ ).encode() ) assert not is_entry_point( textwrap.dedent( """ # Note weird indent. if __name__ == "__main__": main() """ ).encode() ) assert not is_entry_point( textwrap.dedent( """ # Note some nonsense, as a soundness check. print(__name__) """ ).encode() )
def test_is_entry_point_true() -> None: assert is_entry_point( textwrap.dedent( """ # Note single quotes. if __name__ == '__main__': main() """ ).encode() ) assert is_entry_point( textwrap.dedent( """ # Note double quotes. if __name__ == "__main__": main() """ ).encode() ) assert is_entry_point( textwrap.dedent( """ # Note weird extra spaces. if __name__ == "__main__": main() """ ).encode() ) assert is_entry_point( textwrap.dedent( """ # Note trailing comment. if __name__ == "__main__": # Trailing comment. main() """ ).encode() ) assert is_entry_point( textwrap.dedent( """ # Note trailing comment. if __name__ == "__main__":# Trailing comment. main() """ ).encode() ) assert is_entry_point( textwrap.dedent( """ # Note trailing comment. if __name__ == "__main__": # Trailing comment. main() """ ).encode() )