예제 #1
0
def test_get_bucket(buckets, length, expected_bucket):
    bucket = data_io.get_bucket(length, buckets)
    assert bucket == expected_bucket
예제 #2
0
def test_get_bucket(buckets, length, expected_bucket):
    bucket = data_io.get_bucket(length, buckets)
    assert bucket == expected_bucket
예제 #3
0
def test_get_bucket(buckets, length, expected_bucket):
    pytest.importorskip('mxnet')
    from sockeye import data_io
    bucket = data_io.get_bucket(length, buckets)
    assert bucket == expected_bucket