def test_TwoBandsTwoBitsEncoder_init():
'''
Test dual bit two band encoder constructor
'''
encoder = TwoBandsTwoBitsEncoder(0, 1)
assert encoder.l1Index == 0
assert encoder.l2Index == 1
encoder = TwoBandsTwoBitsEncoder(1, 0)
assert encoder.l1Index == 1
assert encoder.l2Index == 0
def test_TwoBandsTwoBitsEncoder_addSamples1():
'''
Test dual bit encoder samples adding and conversion
'''
encoder = TwoBandsTwoBitsEncoder(0, 1)
samples = numpy.ndarray((2, 2), 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 == 8
def test_TwoBandsTwoBitsEncoder_addSamples0():
'''
Test dual bit encoder samples adding and conversion
'''
encoder = TwoBandsTwoBitsEncoder(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 == 0x22).all()
converted = encoder.flush()
assert len(converted) == 1
assert converted.dtype == numpy.uint8
assert converted[0] == 0x20
