johnmee / codility Goto Github PK

from math import ceil
import random
import unittest
MAXLENGHT = 100000
MINLENGHT = 3
RANGEVALUE = (-2147483648,2147483648)
def solution(A):
    
    if not isinstance(A, list):
        raise TypeError("Input A must be a List.")
        
    if not len(A):
        return 0
    
    if len(A) < MINLENGHT:
        return 0

    if len(A) > MAXLENGHT:
        raise ValueError(f"Input A should not be greater than {MAXLENGHT}")
    
    A.sort()
    for i in range(len(A) - 2):
        if A[i] > 0 and A[i] <= A[i+1] and (A[i]+A[i+1]) > A[i+2]:
            return 1
    return 0

class TestExercise(unittest.TestCase):

    def test_simple(self):
        self.assertEqual(solution([-100, 2, 4, 5]), 1)
        self.assertEqual(solution([2,3,5]), 0)
        self.assertEqual(solution([3,3,5]), 1)
        self.assertEqual(solution([2,2,2]), 1)
        self.assertNotEqual(solution([1,2,3]), 1)
        self.assertEqual(solution([10,20,30]), 0)
        self.assertEqual(solution([10,11,20]), 1)
        
    def test_type(self):
        self.assertRaises(TypeError, solution, {})
        self.assertRaises(TypeError, solution, "5,5,5")

    def test_value(self):
        self.assertRaises(ValueError, solution, [1,1]*MAXLENGHT)

    def test_extreme(self):
        print(solution([random.randint(RANGEVALUE[0], RANGEVALUE[1]) for _ in range(MAXLENGHT)]))
        print(solution([random.randint(RANGEVALUE[0], -1) for _ in range(MAXLENGHT)]))
        print(solution([random.randint(-10, -1) for _ in range(MAXLENGHT)]))
        
if __name__ == '__main__':
    unittest.main()