def testMapBucketIndexToNonZeroBits(self):
"""
Test that mapBucketIndexToNonZeroBits works and that max buckets and
clipping are handled properly.
"""
enc = RandomDistributedScalarEncoder(resolution=1.0, w=11, n=150)
# Set a low number of max buckets
enc._initializeBucketMap(10, None)
enc.encode(0.0)
enc.encode(-7.0)
enc.encode(7.0)
self.assertEqual(len(enc.bucketMap), enc._maxBuckets,
"_maxBuckets exceeded")
self.assertTrue(
(enc.mapBucketIndexToNonZeroBits(-1) == enc.bucketMap[0]).all(),
"mapBucketIndexToNonZeroBits did not handle negative index")
self.assertTrue(
(enc.mapBucketIndexToNonZeroBits(1000) == enc.bucketMap[9]).all(),
"mapBucketIndexToNonZeroBits did not handle negative index")
e23 = enc.encode(23.0)
e6 = enc.encode(6)
self.assertEqual((e23 == e6).sum(), enc.getWidth(),
"Values not clipped correctly during encoding")
e_8 = enc.encode(-8)
e_7 = enc.encode(-7)
self.assertEqual((e_8 == e_7).sum(), enc.getWidth(),
"Values not clipped correctly during encoding")
self.assertEqual(enc.getBucketIndices(-8)[0], 0,
"getBucketIndices returned negative bucket index")
self.assertEqual(enc.getBucketIndices(23)[0], enc._maxBuckets-1,
"getBucketIndices returned bucket index that is too large")
def testMapBucketIndexToNonZeroBits(self):
"""
Test that mapBucketIndexToNonZeroBits works and that max buckets and
clipping are handled properly.
"""
encoder = RandomDistributedScalarEncoder(resolution=1.0, w=11, n=150)
# Set a low number of max buckets
encoder._initializeBucketMap(10, None)
encoder.encode(0.0)
encoder.encode(-7.0)
encoder.encode(7.0)
self.assertEqual(len(encoder.bucketMap), encoder._maxBuckets,
"_maxBuckets exceeded")
self.assertTrue(
numpy.array_equal(encoder.mapBucketIndexToNonZeroBits(-1),
encoder.bucketMap[0]),
"mapBucketIndexToNonZeroBits did not handle negative"
" index")
self.assertTrue(
numpy.array_equal(encoder.mapBucketIndexToNonZeroBits(1000),
encoder.bucketMap[9]),
"mapBucketIndexToNonZeroBits did not handle negative index")
e23 = encoder.encode(23.0)
e6 = encoder.encode(6)
self.assertEqual((e23 == e6).sum(), encoder.getWidth(),
"Values not clipped correctly during encoding")
ep8 = encoder.encode(-8)
ep7 = encoder.encode(-7)
self.assertEqual((ep8 == ep7).sum(), encoder.getWidth(),
"Values not clipped correctly during encoding")
self.assertEqual(
encoder.getBucketIndices(-8)[0], 0,
"getBucketIndices returned negative bucket index")
self.assertEqual(
encoder.getBucketIndices(23)[0], encoder._maxBuckets - 1,
"getBucketIndices returned bucket index that is too"
" large")
