Added support for restrict, total equality for countPET tests.