Update assembly cache tests to PyOP2 changes