Remove remaining py.test imports in favor of Assay