def test_dataset_conversion_transpose_regression(self):
qd = DataSet2D(name="test")
qd.x_set.label = "X label"
qd.x_set.unit = "seconds"
ds = qd.to_xarray().transpose()
qdt = DataSet.from_xarray(ds)
self.assertEqual(qd.x_set.label, qdt.x_set.label)
self.assertEqual(qd.x_set.unit, qdt.x_set.unit)
self.assertEqual([a.name for a in qdt.z.set_arrays],
["y_set", "x_set"])
def test_xarray_example_conversion(self):
times = pd.date_range("2000-01-01", "2000-1-31", name="time")
shape = (31, 3)
xarray_dataset = xr.Dataset(
{"tmin": (("time", "location"), np.random.rand(*shape)),
"tmax": (("time", "location"), np.random.rand(*shape)),
}, {"time": times, "location": ["IA", "IN", "IL"]},)
qd = DataSet.from_xarray(xarray_dataset)
xarray_dataset2 = qd.to_xarray()
self.assertEqual(qd.default_parameter_array().shape, xarray_dataset.tmin.shape)
self.assertEqual(list(xarray_dataset.coords.keys()), list(xarray_dataset2.coords.keys()))
self.assertEqual(list(xarray_dataset.data_vars.keys()), list(xarray_dataset2.data_vars.keys()))
self.assertEqual(xarray_dataset.tmin.shape, xarray_dataset2.tmin.shape)
def test_xarray_conversions(self):
qd = DataSet1D(name="TestNewData_test_xarray_conversions")
xarray_data_set = qcodes_dataset_to_xarray_dataset(qd)
qd_transformed = xarray_dataset_to_qcodes_dataset(xarray_data_set)
m = qd.default_parameter_array()
mt = qd_transformed.default_parameter_array()
for key in ["name", "unit"]:
self.assertEqual(getattr(m, key), getattr(mt, key))
qd2 = DataSet2D(name="TestNewData_test_xarray_conversions")
xarray_data_set = qcodes_dataset_to_xarray_dataset(qd2)
qd2_transformed = xarray_dataset_to_qcodes_dataset(xarray_data_set)
m = qd2.default_parameter_array()
mt = qd2_transformed.default_parameter_array()
for key in ["name", "unit"]:
self.assertEqual(getattr(m, key), getattr(mt, key))
xds = qd.to_xarray()
qds = DataSet.from_xarray(xds)
