def test_get_in_and_out_degrees_bad_mod_throws(self):
with self.assertRaises(ValueError) as context:
get_in_and_out_degrees(
[['1', '2', '4'], ['2', '4', '2'], ['1', '2', '4']], 7)
self.assertEqual(
'The number of total accounts (7) must be a multiple of the degree sequence length (4).',
str(context.exception))
def test_get_in_and_out_degrees_unequal_degrees_throws(self):
with self.assertRaises(ValueError) as context:
get_in_and_out_degrees(
[['1', '2', '4'], ['2', '4', '2'], ['1', '2', '3']], 8)
self.assertEqual(
'The sum of in-degree (12) and out-degree (11) must be same.',
str(context.exception))
def test_get_in_and_out_degrees_padding(self):
result = get_in_and_out_degrees(
[['1', '2', '4'], ['2', '4', '2'], ['1', '2', '4']], 8)
self.assertEqual(result,
([2, 4, 4, 2, 2, 4, 4, 2], [4, 2, 2, 4, 4, 2, 2, 4]))