def test_BandTwoBitsEncoder_init():
'''
Test dual bit encoder constructor
'''
encoder = BandTwoBitsEncoder(0)
assert encoder.bandIndex == 0
encoder = BandTwoBitsEncoder(1)
assert encoder.bandIndex == 1
def test_BandTwoBitsEncoder_addSamples1():
'''
Test dual bit encoder samples adding and conversion
'''
encoder = BandTwoBitsEncoder(0)
samples = numpy.ndarray((1, 2), dtype=numpy.float)
samples[0][1::2].fill(-1.)
samples[0][0::2].fill(1.)
converted = encoder.addSamples(samples)
assert len(converted) == 0
assert converted.dtype == numpy.uint8
assert encoder.n_bits == 4
def test_BandTwoBitsEncoder_addSamples1(): ''' Test dual bit encoder samples adding and conversion ''' encoder = BandTwoBitsEncoder(0) samples = numpy.ndarray((1, 2), dtype=numpy.float) samples[0][1::2].fill(-1.) samples[0][0::2].fill(1.) converted = encoder.addSamples(samples) assert len(converted) == 0 assert converted.dtype == numpy.uint8 assert encoder.n_bits == 4
def test_BandTwoBitsEncoder_addSamples0():
'''
Test dual bit encoder samples adding and conversion
'''
encoder = BandTwoBitsEncoder(0)
samples = numpy.ndarray((1, EncoderBase.BLOCK_SIZE + 3), dtype=numpy.float)
samples[0][1::2].fill(-1.)
samples[0][0::2].fill(1.)
converted = encoder.addSamples(samples)
assert len(converted) == len(samples[0]) / 4
assert converted.dtype == numpy.uint8
assert (converted == 0x88).all()
converted = encoder.flush()
assert len(converted) == 1
assert converted.dtype == numpy.uint8
assert converted[0] == 0x88
def test_BandTwoBitsEncoder_addSamples0(): ''' Test dual bit encoder samples adding and conversion ''' encoder = BandTwoBitsEncoder(0) samples = numpy.ndarray((1, EncoderBase.BLOCK_SIZE + 3), dtype=numpy.float) samples[0][1::2].fill(-1.) samples[0][0::2].fill(1.) converted = encoder.addSamples(samples) assert len(converted) == len(samples[0]) / 4 assert converted.dtype == numpy.uint8 assert (converted == 0x88).all() converted = encoder.flush() assert len(converted) == 1 assert converted.dtype == numpy.uint8 assert converted[0] == 0x88
def test_BandTwoBitsEncoder_convertBand0():
'''
Test dual bit encoder band conversion
'''
samples = numpy.ndarray(10, dtype=numpy.float)
samples[1::2].fill(-1.)
samples[0::2].fill(1.)
signs, amps = BandTwoBitsEncoder.convertBand(samples)
assert signs.dtype == numpy.bool
assert amps.dtype == numpy.bool
assert len(signs) == len(samples)
assert len(amps) == len(samples)
assert (amps == 0).all()
assert (signs[1::2] == 0).all()
assert (signs[0::2] == 1).all()
def test_BandTwoBitsEncoder_convertBand0(): ''' Test dual bit encoder band conversion ''' samples = numpy.ndarray(10, dtype=numpy.float) samples[1::2].fill(-1.) samples[0::2].fill(1.) signs, amps = BandTwoBitsEncoder.convertBand(samples) assert signs.dtype == numpy.bool assert amps.dtype == numpy.bool assert len(signs) == len(samples) assert len(amps) == len(samples) assert (amps == 0).all() assert (signs[1::2] == 0).all() assert (signs[0::2] == 1).all()
