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_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
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