def test_has_model(self) -> None: itq = ItqFunctor() # with no vector/rotation set, should return false. self.assertFalse(itq.has_model()) # If only one of the two is None, then false should be returned. itq.mean_vec = 'mean vec' itq.rotation = None self.assertFalse(itq.has_model()) itq.mean_vec = None itq.rotation = 'rotation' self.assertFalse(itq.has_model()) # If both are not None, return true. itq.mean_vec = 'mean vec' itq.rotation = 'rotation' self.assertTrue(itq.has_model())
def test_get_hash(self) -> None: fit_descriptors = [] for i in range(5): d = DescriptorMemoryElement('test', i) d.set_vector([-2. + i, -2. + i]) fit_descriptors.append(d) # The following "rotation" matrix should cause any 2-feature descriptor # to the right of the line ``y = -x`` to be True, and to the left as # False. If on the line, should be True. itq = ItqFunctor(bit_length=1, random_seed=0) itq.mean_vec = numpy.array([0., 0.]) itq.rotation = numpy.array([[1. / sqrt(2)], [1. / sqrt(2)]]) numpy.testing.assert_array_equal(itq.get_hash(numpy.array([1, 1])), [True]) numpy.testing.assert_array_equal(itq.get_hash(numpy.array([-1, -1])), [False]) numpy.testing.assert_array_equal(itq.get_hash(numpy.array([-1, 1])), [True]) numpy.testing.assert_array_equal( itq.get_hash(numpy.array([-1.001, 1])), [False]) numpy.testing.assert_array_equal( itq.get_hash(numpy.array([-1, 1.001])), [True]) numpy.testing.assert_array_equal(itq.get_hash(numpy.array([1, -1])), [True]) numpy.testing.assert_array_equal( itq.get_hash(numpy.array([1, -1.001])), [False]) numpy.testing.assert_array_equal( itq.get_hash(numpy.array([1.001, -1])), [True])
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_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_fit_has_model(self) -> None: # When trying to run fit where there is already a mean vector and # rotation set. itq = ItqFunctor() itq.mean_vec = 'sim vec' itq.rotation = 'sim rot' self.assertRaisesRegex(RuntimeError, "Model components have already been loaded.", itq.fit, [])
def test_save_model_no_caches(self) -> None: expected_mean_vec = numpy.array([1, 2, 3]) expected_rotation = numpy.eye(3) # Cache variables should remain None after save. itq = ItqFunctor() itq.mean_vec = expected_mean_vec itq.rotation = expected_rotation itq.save_model() self.assertIsNone(itq.mean_vec_cache_elem) self.assertIsNone(itq.mean_vec_cache_elem)