def test_shorter_gen_and_list(self): """Test shorter function on generators and lists """ self.assertTrue(shorter(range(2), [1, 2, 3])) self.assertFalse(shorter([1, 2, 3, 4], range(2))) self.assertTrue(shorter([0], count())) self.assertFalse(shorter(count(), [1, 2, 3]))
def test_shorter_generators(self): """Test shorter function on generators """ self.assertTrue(shorter(range(10), range(20))) self.assertFalse(shorter(range(3), range(2))) self.assertTrue(shorter(range(20), count())) self.assertFalse(shorter(count(), range(20)))
def test_shorter_lists(self): """Test shorter function on lists """ self.assertTrue(shorter([], [1])) self.assertTrue(shorter([1, 2], [1, 2, 3])) self.assertFalse(shorter([1], [])) self.assertFalse(shorter([1, 2, 3], [1, 2]))