py2mat: xarray

Author: scivision

.gitignore

@@ -1 +1,2 @@
 *.pkl
+*.asv

TestPy2Mat.m

@@ -70,6 +70,23 @@ function test_py2mat_scalar(tc, matlabS)
 
 end
 
+
+function test_xarray_dataarray(tc)
+
+t0 = datetime(2021, 12, 25, 1, 2, 34.56);
+A = pi;
+
+coords = dictionary("time", t0);
+
+% https://docs.xarray.dev/en/latest/generated/xarray.DataArray.html
+pyDataArray = py.xarray.DataArray(A, coords);
+
+B = py2mat(pyDataArray);
+
+tc.verifyEqual(B, A);
+
+end
+
 end
 
 end

buildfile.m

@@ -1,7 +1,6 @@
 function plan = buildfile
 plan = buildplan(localfunctions);
 plan.DefaultTasks = "test";
-plan("test").Dependencies = "check";
 end
 
 function checkTask(~)
@@ -11,7 +10,7 @@ function checkTask(~)
 end
 
 function testTask(context)
-r = runtests(context.Plan.RootFolder, strict=true);
+r = runtests(context.Plan.RootFolder, IncludeSubfolders=true, strict=true);
 assert(~isempty(r), "No tests were run")
 assertSuccess(r)
 end

py2mat.m

@@ -194,6 +194,9 @@
           x_mat.(string(k{1})) = py2mat(x_py.get(k{1}));
       end
 
+    case 'py.xarray.core.dataarray.DataArray'
+      x_mat = double(py.numpy.asfortranarray(x_py));
+
     % punt
     otherwise
       % return the original item?  nothing?
@@ -202,3 +205,24 @@
       x_mat = [];
   end
 end
+
+
+function I = xarrayind2vector(V, key) %#ok<DEFNU>
+%% XARRAYIND2VECTOR convert xarray Index to Matlab vector array
+% by SciVision
+
+C = cell(V.indexes{key}.values.tolist);
+% might be numeric or cell array of strings
+
+if iscellstr(C) || (length(C) > 1 && isa(C{1}, 'py.str'))
+  I = cellfun(@char, C, 'uniformoutput', false);
+elseif isa(C{1}, 'py.datetime.datetime')
+  I = char(C{1}.isoformat());
+elseif isa(C{1}, 'py.int') && C{1} > 1e12
+  C = cellfun(@double,C)/1e9;
+  I = datetime(C, 'convertfrom','posixtime');
+else
+  I = cell2mat(C);
+end
+
+end