def METHOD_NAME(self):
    for dtype in TEST_DTYPES:
        embedding_angles = [0, 20, 40, 60, 80]
        embeddings = torch.tensor(
            [angle_to_coord(a) for a in embedding_angles],
            requires_grad=True,
            dtype=dtype,
        ).to(
            TEST_DEVICE
        )  # 2D embeddings
        labels = torch.LongTensor([0, 0, 1, 1, 2])
        pos_pairs = [(0, 1), (1, 0), (2, 3), (3, 2)]
        neg_pairs = [
            (0, 2),
            (0, 3),
            (0, 4),
            (1, 2),
            (1, 3),
            (1, 4),
            (2, 0),
            (2, 1),
            (2, 4),
            (3, 0),
            (3, 1),
            (3, 4),
            (4, 0),
            (4, 1),
            (4, 2),
            (4, 3),
        ]
        self.helper(embeddings, labels, pos_pairs, neg_pairs, dtype)