def test_none_set(self) -> None: d = DescriptorMemoryElement(0) self.assertFalse(d.has_vector()) d.set_vector(numpy.ones(16)) self.assertTrue(d.has_vector()) numpy.testing.assert_equal(d.vector(), numpy.ones(16)) d.set_vector(None) self.assertFalse(d.has_vector()) self.assertIs(d.vector(), None)
def test_output_immutability(self) -> None: # make sure that data stored is not susceptible to modifications after # extraction v = numpy.ones(16) d = DescriptorMemoryElement(0) self.assertFalse(d.has_vector()) d.set_vector(v) r = d.vector() assert r is not None r[:] = 0 self.assertEqual(r.sum(), 0) r_again = d.vector() assert r_again is not None self.assertEqual(r_again.sum(), 16)