def test_symetric_attention(self):
    in_features = self.x1.shape[1]
    out_features = self.x1.shape[1]
    att_layer = SpGraphAttentionLayer(in_features, out_features, self.opt, self.device, concat=True)
    (attention, _) = att_layer(self.x1, self.edge1)
    self.assertTrue(torch.all(torch.eq(attention, (0.5 * torch.ones((self.edge1.shape[1], self.x1.shape[1]))))))
def test_symetric_attention(self):
    in_features = self.x1.shape[1]
    out_features = self.x1.shape[1]
    att_layer = SpGraphAttentionLayer(in_features, out_features, self.opt, self.device, concat=True)
    (attention, _) = att_layer(self.x1, self.edge1)
    self.assertTrue(torch.all(torch.eq(attention, (0.5 * torch.ones((self.edge1.shape[1], self.x1.shape[1]))))))