class DeltaEncoderTest(unittest.TestCase):
"""Unit tests for DeltaEncoder class"""
def setUp(self):
self._dencoder = DeltaEncoder(w=21, n=100, forced=True)
self._adaptscalar = AdaptiveScalarEncoder(w=21, n=100, forced=True)
def testDeltaEncoder(self):
"""simple delta reconstruction test"""
for i in range(5):
encarr = self._dencoder.encodeIntoArray(i, np.zeros(100), learn=True)
self._dencoder.setStateLock(True)
for i in range(5, 7):
encarr = self._dencoder.encodeIntoArray(i, np.zeros(100), learn=True)
res = self._dencoder.topDownCompute(encarr)
self.assertEqual(res[0].value, 6)
self.assertEqual(self._dencoder.topDownCompute(encarr)[0].value,
res[0].value)
self.assertEqual(self._dencoder.topDownCompute(encarr)[0].scalar,
res[0].scalar)
self.assertTrue(np.array_equal(
self._dencoder.topDownCompute(encarr)[0].encoding,
res[0].encoding))
def testEncodingVerification(self):
"""encoding verification test passed"""
feedIn = [1, 10, 4, 7, 9, 6, 3, 1]
expectedOut = [0, 9, -6, 3, 2, -3, -3, -2]
self._dencoder.setStateLock(False)
#Check that the deltas are being returned correctly.
for i in range(len(feedIn)):
aseencode = np.zeros(100)
self._adaptscalar.encodeIntoArray(expectedOut[i], aseencode, learn=True)
delencode = np.zeros(100)
self._dencoder.encodeIntoArray(feedIn[i], delencode, learn=True)
self.assertTrue(np.array_equal(delencode[0], aseencode[0]))
def testLockingState(self):
"""Check that locking the state works correctly"""
feedIn = [1, 10, 9, 7, 9, 6, 3, 1]
expectedOut = [0, 9, -6, 3, 2, -3, -3, -2]
for i in range(len(feedIn)):
if i == 3:
self._dencoder.setStateLock(True)
aseencode = np.zeros(100)
self._adaptscalar.encodeIntoArray(expectedOut[i], aseencode, learn=True)
delencode = np.zeros(100)
if i>=3:
self._dencoder.encodeIntoArray(feedIn[i]-feedIn[2], delencode,
learn=True)
else:
self._dencoder.encodeIntoArray(expectedOut[i], delencode, learn=True)
self.assertTrue(np.array_equal(delencode[0], aseencode[0]))
def testEncodeInvalidInputType(self):
try:
self._dencoder.encode("String")
except TypeError as e:
self.assertEqual(
e.message,
"Expected a scalar input but got input of type <type 'str'>")
else:
self.fail("Should have thrown TypeError during attempt to encode string "
"with scalar encoder.")
def testReadWrite(self):
feedIn = [1, 10, 4, 7, 9, 6, 3, 1]
expectedOut = [0, 9, -6, 3, 2, -3, -3, -2]
self._dencoder.setStateLock(False)
outp = []
#Check that the deltas are being returned correctly.
for i in range(len(feedIn)-1):
aseencode = np.zeros(100)
self._adaptscalar.encodeIntoArray(expectedOut[i], aseencode, learn=True)
delencode = np.zeros(100)
self._dencoder.encodeIntoArray(feedIn[i], delencode, learn=True)
outp.append(delencode)
proto1 = DeltaEncoderProto.new_message()
self._dencoder.write(proto1)
# Write the proto to a temp file and read it back into a new proto
with tempfile.TemporaryFile() as f:
proto1.write(f)
f.seek(0)
proto2 = DeltaEncoderProto.read(f)
encoder = DeltaEncoder.read(proto2)
self.assertIsInstance(encoder, DeltaEncoder)
self.assertEqual(encoder.width, self._dencoder.width)
self.assertEqual(encoder.n, self._dencoder.n)
self.assertEqual(encoder.name, self._dencoder.name)
self.assertEqual(encoder._prevAbsolute, self._dencoder._prevAbsolute)
self.assertEqual(encoder._prevDelta, self._dencoder._prevDelta)
self.assertEqual(encoder._stateLock, self._dencoder._stateLock)
delencode = np.zeros(100)
self._dencoder.encodeIntoArray(feedIn[-1], delencode, learn=True)
delencode2 = np.zeros(100)
encoder.encodeIntoArray(feedIn[-1], delencode2, learn=True)
self.assertTrue(np.array_equal(delencode, delencode2))
