def test_get_config_no_cache(self):
itq = ItqFunctor(bit_length=1, itq_iterations=2, normalize=3,
random_seed=4)
c = itq.get_config()
NT.assert_equal(c['bit_length'], 1)
NT.assert_equal(c['itq_iterations'], 2)
NT.assert_equal(c['normalize'], 3)
NT.assert_equal(c['random_seed'], 4)
NT.assert_is_none(c['mean_vec_cache']['type'])
NT.assert_is_none(c['rotation_cache']['type'])
def test_get_config_no_cache(self):
itq = ItqFunctor(bit_length=1, itq_iterations=2, normalize=3,
random_seed=4)
c = itq.get_config()
self.assertEqual(c['bit_length'], 1)
self.assertEqual(c['itq_iterations'], 2)
self.assertEqual(c['normalize'], 3)
self.assertEqual(c['random_seed'], 4)
self.assertIsNone(c['mean_vec_cache']['type'])
self.assertIsNone(c['rotation_cache']['type'])
def test_get_config_with_cache_elements(self):
itq = ItqFunctor(bit_length=5, itq_iterations=6, normalize=7,
random_seed=8)
itq.mean_vec_cache_elem = DataMemoryElement('cached vec bytes')
itq.rotation_cache_elem = DataMemoryElement('cached rot bytes')
c = itq.get_config()
NT.assert_equal(c['bit_length'], 5)
NT.assert_equal(c['itq_iterations'], 6)
NT.assert_equal(c['normalize'], 7)
NT.assert_equal(c['random_seed'], 8)
NT.assert_equal(c['mean_vec_cache']['type'], "DataMemoryElement")
NT.assert_equal(c['mean_vec_cache']['DataMemoryElement']['bytes'],
'cached vec bytes')
NT.assert_equal(c['rotation_cache']['DataMemoryElement']['bytes'],
'cached rot bytes')
def test_get_config_with_cache_elements(self):
itq = ItqFunctor(bit_length=5, itq_iterations=6, normalize=7,
random_seed=8)
itq.mean_vec_cache_elem = DataMemoryElement('cached vec bytes')
itq.rotation_cache_elem = DataMemoryElement('cached rot bytes')
c = itq.get_config()
self.assertEqual(c['bit_length'], 5)
self.assertEqual(c['itq_iterations'], 6)
self.assertEqual(c['normalize'], 7)
self.assertEqual(c['random_seed'], 8)
self.assertEqual(c['mean_vec_cache']['type'], "DataMemoryElement")
self.assertEqual(c['mean_vec_cache']['DataMemoryElement']['bytes'],
'cached vec bytes')
self.assertEqual(c['rotation_cache']['DataMemoryElement']['bytes'],
'cached rot bytes')
def test_get_config_with_cache_elements(self):
itq = ItqFunctor(bit_length=5, itq_iterations=6, normalize=7,
random_seed=8)
itq.mean_vec_cache_elem = DataMemoryElement(b'cached vec bytes')
itq.rotation_cache_elem = DataMemoryElement(b'cached rot bytes')
c = itq.get_config()
self.assertEqual(c['bit_length'], 5)
self.assertEqual(c['itq_iterations'], 6)
self.assertEqual(c['normalize'], 7)
self.assertEqual(c['random_seed'], 8)
self.assertEqual(c['mean_vec_cache']['type'], "DataMemoryElement")
# Check using string encodings of set bytes (JSON compliant).
self.assertEqual(c['mean_vec_cache']['DataMemoryElement']['bytes'],
'cached vec bytes')
self.assertEqual(c['rotation_cache']['DataMemoryElement']['bytes'],
'cached rot bytes')
