Create test_mask.py