def test_save_model_with_read_only_cache(self) -> None: # If one or both cache elements are read-only, no saving. expected_mean_vec = numpy.array([1, 2, 3]) expected_rotation = numpy.eye(3) itq = ItqFunctor() itq.mean_vec = expected_mean_vec itq.rotation = expected_rotation # read-only mean-vec cache itq.mean_vec_cache_elem = DataMemoryElement(readonly=True) itq.rotation_cache_elem = DataMemoryElement(readonly=False) itq.save_model() self.assertEqual(itq.mean_vec_cache_elem.get_bytes(), b'') self.assertEqual(itq.rotation_cache_elem.get_bytes(), b'') # read-only rotation cache itq.mean_vec_cache_elem = DataMemoryElement(readonly=False) itq.rotation_cache_elem = DataMemoryElement(readonly=True) itq.save_model() self.assertEqual(itq.mean_vec_cache_elem.get_bytes(), b'') self.assertEqual(itq.rotation_cache_elem.get_bytes(), b'') # Both read-only itq.mean_vec_cache_elem = DataMemoryElement(readonly=True) itq.rotation_cache_elem = DataMemoryElement(readonly=True) itq.save_model() self.assertEqual(itq.mean_vec_cache_elem.get_bytes(), b'') self.assertEqual(itq.rotation_cache_elem.get_bytes(), b'')
def test_save_model_with_writable_caches(self) -> None: # If one or both cache elements are read-only, no saving. expected_mean_vec = numpy.array([1, 2, 3]) expected_rotation = numpy.eye(3) expected_mean_vec_bio = BytesIO() # noinspection PyTypeChecker numpy.save(expected_mean_vec_bio, expected_mean_vec) expected_mean_vec_bytes = expected_mean_vec_bio.getvalue() expected_rotation_bio = BytesIO() # noinspection PyTypeChecker numpy.save(expected_rotation_bio, expected_rotation) expected_rotation_bytes = expected_rotation_bio.getvalue() itq = ItqFunctor() itq.mean_vec = expected_mean_vec itq.rotation = expected_rotation itq.mean_vec_cache_elem = DataMemoryElement(readonly=False) itq.rotation_cache_elem = DataMemoryElement(readonly=False) itq.save_model() self.assertEqual(itq.mean_vec_cache_elem.get_bytes(), expected_mean_vec_bytes) self.assertEqual(itq.rotation_cache_elem.get_bytes(), expected_rotation_bytes)
def test_get_config_with_cache_elements(self) -> None: 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) dme_key = 'smqtk_dataprovider.impls.data_element.memory.DataMemoryElement' self.assertEqual(c['mean_vec_cache']['type'], dme_key) # Check using string encodings of set bytes (JSON compliant). self.assertEqual(c['mean_vec_cache'][dme_key]['bytes'], 'cached vec bytes') self.assertEqual(c['rotation_cache'][dme_key]['bytes'], 'cached rot bytes')