clustered_se: Simplify API to just use the fit result object and update test module accordingly