Added get_axes_local_distribution_strategy method.

d2o/distributed_data_object.py

@@ -1760,6 +1760,10 @@ class distributed_data_object(object):
         return self.distributor.consolidate_data(self.data,
                                                  target_rank=target_rank)
 
+    def get_axes_local_distribution_strategy(self, axes):
+        axes = cast_axis_to_tuple(axes, len(self.shape))
+        return self.distributor.get_axes_local_distribution_strategy(axes)
+
     def flatten(self, inplace=False):
         """ Returns a flat copy of the d2o collapsed into one dimension.
 

d2o/distributor_factory.py

@@ -1830,6 +1830,12 @@ class _slicing_distributor(distributor):
     def get_iter(self, d2o):
         return d2o_slicing_iter(d2o)
 
+    def get_axes_local_distribution_strategy(self, axes):
+        if 0 in axes:
+            return self.distribution_strategy
+        else:
+            return 'not'
+
 
 def _equal_slicer(comm, global_shape):
     rank = comm.rank
@@ -2140,3 +2146,6 @@ class _not_distributor(distributor):
 
     def get_iter(self, d2o):
         return d2o_not_iter(d2o)
+
+    def get_axes_local_distribution_strategy(self, axes):
+        return 'not'