Test utils: Narrowed except: to AssertionError
Minor test improvement