[test] Migrate to assume instead of chai