def spec_i(self, i_mock): ''' k, power/bispectrum (p0k, p2k, p4k, bk) of ith mock ''' specdict = self.cat_corr['spec'] # copy cat_corr dictionary cat_corr_i = self.cat_corr.copy() cat_corr_i['catalog']['n_mock'] = i_mock spec_i = Spec(self.type, cat_corr_i, **self.kwargs) spec_i.read() if self.type == 'pk': spec_ell = ''.join(['p', str(specdict['ell']), 'k']) spec_i_spec = getattr(spec_i, spec_ell) return [spec_i.k, spec_i_spec] elif self.type == 'bk': bk_i = getattr(spec_i, 'bk') qk_i = getattr(spec_i, 'qk') return [spec_i.k1, spec_i.k2, spec_i.k3, bk_i, qk_i] else: raise NotImplementedError