Update capacityTest.py

Update to allow dynamic number of loops through the test