def METHOD_NAME(self) -> None:
    before = """
        from potato import union
        union("A", "B")
    """
    after = """
        from potato import union
        union("A", "B")
    """
    self.assertCodemod(before, after, use_pipe_syntax=False)