def METHOD_NAME(self):
    a = torch.arange(7)
    r = unfold(a, 3, 3, 1)
    expected = torch.tensor(
        [
            [0, 1, 2],
            [3, 4, 5],
        ]
    )
    assert_equal(r, expected)
    r = unfold(a, 3, 2, 1)
    expected = torch.tensor([[0, 1, 2], [2, 3, 4], [4, 5, 6]])
    assert_equal(r, expected)
    r = unfold(a, 3, 2, 2)
    expected = torch.tensor(
        [
            [0, 2, 4],
            [2, 4, 6],
        ]
    )
    assert_equal(r, expected)