def test_get_default_bucket_key(buckets, expected_default_bucket_key): default_bucket_key = data_io.get_default_bucket_key(buckets) assert default_bucket_key == expected_default_bucket_key
def test_get_default_bucket_key(buckets, expected_default_bucket_key): default_bucket_key = data_io.get_default_bucket_key(buckets) assert default_bucket_key == expected_default_bucket_key
def test_get_default_bucket_key(buckets, expected_default_bucket_key): pytest.importorskip('mxnet') from sockeye import data_io default_bucket_key = data_io.get_default_bucket_key(buckets) assert default_bucket_key == expected_default_bucket_key