def METHOD_NAME():
    """Test that get_shape works for multisharded legacy disk dataset."""
    # This is the shape of legacy_data_reshard
    num_datapoints = 100
    num_features = 10
    num_tasks = 10
    # legacy_dataset_reshard is a sharded dataset in the legacy format kept
    # around for testing
    current_dir = os.path.dirname(os.path.abspath(__file__))
    data_dir = os.path.join(current_dir, "legacy_dataset_reshard")
    dataset = dc.data.DiskDataset(data_dir)
    # Should now have 10 shards
    assert dataset.get_number_shards() == 10
    X_shape, y_shape, w_shape, ids_shape = dataset.get_shape()
    assert X_shape == (num_datapoints, num_features)
    assert y_shape == (num_datapoints, num_tasks)
    assert w_shape == (num_datapoints, num_tasks)
    assert ids_shape == (num_datapoints,)