Class wrapping itertools.count to provide some additional functionality

Sequence.py

import itertools

class Sequence(object):
    """
    This class wraps itertools.count to provide additional bound means of
    extracting data from the stream
    """
    def __init__(self, start: int=0, step: int=1):
        self._counter = itertools.count(start=start, step=step)

    def just(self, n: int):
        counter = 0
        while counter < n:
            yield next(self._counter)
            counter += 1

    @property
    def just_one(self):
        yield from self.just(1)

    def __iter__(self):
        return self._counter