python中如何在B.py调用A.py中的函数

A.py文件中的内容

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

def add(x,y):
    print('和为：%d'%(x+y))

B.py文件

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

from A import add
add(1, 2)