aboutsummaryrefslogtreecommitdiff
path: root/challenge-260/steven-wilson/python/ch-1.py
blob: 4305bd667b5f966d11333da462ced0af24abc233 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
#!/usr/bin/env python3


def unique_occurences(*elements):
    '''
    >>> unique_occurences(1,2,2,1,1,3)
    1
    >>> unique_occurences(1,2,3)
    0
    >>> unique_occurences(-2,0,1,-2,1,1,0,1,-2,9)
    1
    '''
    if not all(isinstance(elem, int) for elem in elements):
        raise ValueError("Input must consist of integers")

    from collections import Counter

    counts = Counter(elements)
    occurences = counts.values()
    if len(counts) == len(set(occurences)):
        return 1
    else:
        return 0


if __name__ == "__main__":
    import doctest

    doctest.testmod()