forked from numenta/nupic.core-legacy
-
Notifications
You must be signed in to change notification settings - Fork 73
/
eye_test.py
29 lines (22 loc) · 827 Bytes
/
eye_test.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
""" Unit tests for retina encoder. """
import unittest
import os
import numpy as np
from htm.encoders.eye import Eye
class EyeEncoderTest(unittest.TestCase):
""" Unit tests for Eye encoder class. """
def testBasicUsage(self):
eye = Eye()
eye.reset()
FILE=os.path.join('py','tests','encoders','ronja_the_cat.jpg')
eye.new_image(FILE)
eye.scale = 0.5
#eye.center_view()
eye.position = (400,400)
for _ in range(10):
pos,rot,sc = eye.small_random_movement()
(sdrParvo, sdrMagno) = eye.compute(pos,rot,sc)
#eye.plot(delay=500)
print("Sparsity parvo: {}".format(len(eye.parvo_sdr.sparse)/np.product(eye.parvo_sdr.dimensions)))
print("Sparsity magno: {}".format(len(eye.magno_sdr.sparse)/np.product(eye.magno_sdr.dimensions)))
assert(eye.dimensions == (1,1))