def test_initially_empty_name_set(self):
nparam = 3
names = utilities.extend_names_to_match_nparam(names=None,
nparam=nparam)
expected_names = ['$p_{0}$', '$p_{1}$', '$p_{2}$']
self.assertEqual(
names,
expected_names,
msg=str('Names do not match: Expected - {}, Received - {}'.format(
expected_names, names)))
def test_double_entry_name_set(self):
nparam = 3
names = ['aa', 'zz']
names = utilities.extend_names_to_match_nparam(names=names,
nparam=nparam)
expected_names = ['aa', 'zz', '$p_{2}$']
self.assertEqual(
names,
expected_names,
msg=str('Names do not match: Expected - {}, Received - {}'.format(
expected_names, names)))