Exemplo n.º 1
0
    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)
Exemplo n.º 2
0
 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)