def METHOD_NAME(self):
    d = DataEmbedding(5, 128, data=5)
    r = d(torch.rand(2, 10, 5), torch.rand(2, 10, 5))
    self.assertTrue(hasattr(d.temporal_embedding, "month_embed"))
    self.assertEqual(r.shape[2], 128)