def test_draw_point2d_grayscale_third_order(self, dtype, device):
        """Test plotting multiple [x, y] points on a (1, m, n) image."""
        points = torch.tensor([(1, 3), (2, 4)], device=device)
        color = torch.tensor([100], dtype=dtype, device=device)
        img = torch.zeros(1, 8, 8, dtype=dtype, device=device)
        img = draw_point2d(img, points, color)
        for x, y in points:
            assert_close(img[:, y, x], color.to(img.dtype))
Test plotting multiple [x, y] points on a (1, m, n) image.
tests/utils/test_draw.py