Add tests for cache's to_* methods that raise setpoint warning

microsoft · Feb 23, 2021 · 511949d · 511949d
1 parent 00f4bb3
commit 511949d
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/qcodes/tests/dataset/test_dataset_basic.py b/qcodes/tests/dataset/test_dataset_basic.py
@@ -1375,11 +1375,19 @@ def test_write_data_to_text_file_name_exception(tmp_path):
 
 def test_same_setpoint_warning_for_df_and_xarray(different_setpoint_dataset):
 
-    warning_mesage = ('Independent parameter setpoints are not equal. \
-                Check concatenated output carefully.')
+    warning_message = (
+        "Independent parameter setpoints are not equal. "
+        "Check concatenated output carefully."
+    )
 
-    with pytest.warns(UserWarning, match=warning_mesage):
+    with pytest.warns(UserWarning, match=warning_message):
         different_setpoint_dataset.to_pandas_dataframe()
 
-    with pytest.warns(UserWarning, match=warning_mesage):
+    with pytest.warns(UserWarning, match=warning_message):
         different_setpoint_dataset.to_xarray_dataset()
+
+    with pytest.warns(UserWarning, match=warning_message):
+        different_setpoint_dataset.cache.to_pandas_dataframe()
+
+    with pytest.warns(UserWarning, match=warning_message):
+        different_setpoint_dataset.cache.to_xarray_dataset()