Add test for Trio class