def test_single_pair(self):
size = (40, 30)
real = .01 * torch.rand(size)
im = .01 * torch.rand(size)
mem = HRR(data=ComplexTensor(real, im))
k = ComplexTensor(2 * torch.rand(size), 3 * torch.rand(size))
v = ComplexTensor(5 * torch.ones(size), 2 * torch.ones(size))
mem.write(k, v)
restored = mem.read(k)
torch.testing.assert_allclose(restored.real, v.real, rtol=0, atol=.1)
torch.testing.assert_allclose(restored.im, v.im, rtol=0, atol=.1)
restored_phase = mem.get_phase(k)
torch.testing.assert_allclose(restored_phase, v.phase, rtol=0, atol=.1)
def test_multiple_pairs(self):
size = (40, 30)
real = .01 * torch.rand(size)
im = .01 * torch.rand(size)
mem = HRR(ComplexTensor(real, im))
k = ComplexTensor(torch.rand(size), torch.rand(size))
v = ComplexTensor(5 * torch.ones(size), 2 * torch.ones(size))
mem.write(k, v)
k = ComplexTensor(torch.rand(size), torch.rand(size))
v = ComplexTensor(5 * torch.ones(size), 2 * torch.ones(size))
mem.write(k, v)
k = ComplexTensor(torch.rand(size), torch.rand(size))
v = ComplexTensor(5 * torch.ones(size), 2 * torch.ones(size))
mem.write(k, v)
restored_phase = mem.get_phase(k)
torch.testing.assert_allclose(restored_phase, v.phase, rtol=0, atol=.1)