update capacityTest.py script