adapt canonical_test.py to new api