I located the test.py code on the same level as the my_project folder. I found that in addition to the following entry suggested by @bsraskr.

my_project/__init__.py

from . import draw

You also need:

my_project/draw/__init__.py

from .shapes import box