update test_AtomMask.py