补全Python中实现存取款功能的BankAccount类代码

存取款操作

创建一个名为BankAccount的类，要求：

1. 有构造函数__init__，接受账户名和初始余额；
2. 有deposit(amount)方法用于存款；
3. 有withdraw(amount)方法用于取款（余额不足时提示）；
4. 有display()方法显示账户信息；
5. 创建一个实例，进行存款、取款操作，并显示最终账户信息。

请根据要求，补全Python代码：

class BankAccount:
    def   ①   (self, account_name, initial_balance):
        self.account_name = account_name
        self.balance =    ②    

    def deposit(self, amount):
        if amount > 0:
              ③
              print(f"成功存入{amount}元")
        else:
            print("存款金额必须大于0")

    def withdraw(self, amount):
        if amount > self.balance:
            print("余额不足")
        else:
               ④
              print(f"成功取出{amount}元")

    def display(self):
        print(f"账户名：{self.account_name}")
        print(f"当前余额：{   ⑤   }元")

account = BankAccount("张三", 1000)
account.deposit(500)
account.withdraw(200)
account.withdraw(2000)   # 测试余额不足情况
account.display()