def METHOD_NAME(self):
    pk = cupy.asarray([[0.1, 0.2], [0.6, 0.3], [0.3, 0.5]])
    qk = cupy.asarray([[0.2, 0.1], [0.3, 0.6], [0.5, 0.3]])
    testing.assert_array_almost_equal(
        stats.entropy(pk, qk), [0.1933259, 0.18609809]
    )