def test_FourBandsBitEncoder_addSamples1(): ''' Test single bit encoder samples adding and conversion ''' encoder = FourBandsBitEncoder(0, 1, 0, 1) samples = numpy.ndarray((2, 1), dtype=numpy.float) samples[0].fill(-1.) samples[1].fill(1.) converted = encoder.addSamples(samples) assert len(converted) == 0 assert converted.dtype == numpy.uint8 assert encoder.n_bits == 4
def test_FourBandsBitEncoder_init(): ''' Test single bit four band encoder constructor ''' encoder = FourBandsBitEncoder(0, 1, 2, 3) assert encoder.bandIndexes[0] == 0 assert encoder.bandIndexes[1] == 1 assert encoder.bandIndexes[2] == 2 assert encoder.bandIndexes[3] == 3 encoder = FourBandsBitEncoder(3, 2, 1, 0) assert encoder.bandIndexes[0] == 3 assert encoder.bandIndexes[1] == 2 assert encoder.bandIndexes[2] == 1 assert encoder.bandIndexes[3] == 0
def test_FourBandsBitEncoder_addSamples0(): ''' Test single bit encoder samples adding and conversion ''' encoder = FourBandsBitEncoder(0, 1, 0, 1) samples = numpy.ndarray((2, EncoderBase.BLOCK_SIZE + 1), dtype=numpy.float) samples[0].fill(-1.) samples[1].fill(1.) converted = encoder.addSamples(samples) assert len(converted) == len(samples[0]) / 2 assert converted.dtype == numpy.uint8 assert (converted == 0xAA).all() converted = encoder.flush() assert len(converted) == 1 assert converted.dtype == numpy.uint8 assert converted[0] == 0xA0