Data sampling for latency measurement

sampling.py

import random


class Sampling(object):
    def __init__(self, size_max):
        self.__count = 0
        self.__samples = list()
        self.__size_max = size_max

    def sample(self, value):
        self.__count += 1
        if self.__count < len(self.__samples) or len(self.__samples) < self.__size_max:
            self.__samples.append(value)
        else:
            index = random.randint(0, self.__count - 1)
            if index < len(self.__samples):
                self.__samples[index] = value

    def get_samples(self):
        return self.__samples


def test(count):
    sampling = Sampling(10)
    values = list()
    for val in range(0, count):
        values.append(val)
        sampling.sample(val)
    samples = sampling.get_samples()
    print('values={}'.format(values))
    print('samples={}\n'.format(samples))


if __name__ == "__main__":
    test(5)
    test(10)
    test(20)
    test(100)
    test(1000)