def test_cycle_length(self):
self.assertEqual(my_module.cycle_length(7), 6)
self.assertEqual(my_module.cycle_length(17), 16)
self.assertEqual(my_module.cycle_length(8), 0)
import my_module
lengths = []
for d in range(2, 1001):
lengths.append(my_module.cycle_length(d))
print(lengths.index(max(lengths)) + 2)
