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)