def test_split_param_vec_with_intercepts(self): """ Ensures that split_param_vec returns (shapes, intercepts, index_coefs) when called from within uneven.py. """ # Store the results of split_param_vec() split_results = uneven.split_param_vec(self.fake_all_params, self.fake_rows_to_alts, self.fake_design) # Check for expected results. for item in split_results: self.assertIsInstance(item, np.ndarray) self.assertEqual(len(item.shape), 1) npt.assert_allclose(split_results[0], self.fake_shapes) npt.assert_allclose(split_results[1], self.fake_intercepts) npt.assert_allclose(split_results[2], self.fake_betas) return None
def test_split_param_vec_without_intercepts(self): """ Ensures that split_param_vec returns (shapes, intercepts, index_coefs) when called from within uneven.py. """ # Store the results of split_param_vec() shapes_and_betas = np.concatenate([self.fake_shapes, self.fake_betas]) split_results = uneven.split_param_vec(shapes_and_betas, self.fake_rows_to_alts, self.fake_design) # Check for expected results. for idx in [0, 2]: self.assertIsInstance(split_results[idx], np.ndarray) self.assertEqual(len(split_results[idx].shape), 1) npt.assert_allclose(split_results[0], self.fake_shapes) npt.assert_allclose(split_results[2], self.fake_betas) self.assertIsNone(split_results[1]) return None