def METHOD_NAME(self):
    for dtype in TEST_DTYPES:
        embeddings = torch.arange(6).type(dtype).to(TEST_DEVICE).unsqueeze(1)
        a1, p, a2, n = self.dist_miner(embeddings, self.labels)
        pos_pairs = torch.stack([a1, p], dim=1)
        neg_pairs = torch.stack([a2, n], dim=1)
        self.helper(pos_pairs, neg_pairs)