def METHOD_NAME(test_case, device, shape, low, high):
    for dtype in [
        flow.uint8,
        flow.int8,
        flow.int32,
        flow.int64,
        flow.float32,
        flow.float64,
    ]:
        y = flow.randint(low, high, shape, dtype=dtype, device=flow.device(device))
        test_case.assertTrue(y.dtype == dtype)
        test_case.assertTrue(y.shape == shape)