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')