Test updates for basic.py