|
6 | 6 | import unittest
|
7 | 7 | from datetime import datetime, timezone
|
8 | 8 |
|
9 |
| -from ddt import data, ddt, unpack |
10 |
| - |
| 9 | +import numpy as np |
11 | 10 | import speasy as spz
|
| 11 | +from ddt import data, ddt, unpack |
| 12 | +from speasy.config import amda as amda_cfg |
12 | 13 | from speasy.inventories import flat_inventories
|
13 | 14 | from speasy.products import SpeasyVariable
|
14 | 15 | from speasy.webservices.amda import ProductType
|
@@ -253,6 +254,16 @@ def test_raises_if_user_passes_unknown_product_kwargs_to_get_data(self):
|
253 | 254 | with self.assertRaises(ValueError):
|
254 | 255 | spz.get_data('amda/This_product_does_not_exist', "2018-01-01", "2018-01-02")
|
255 | 256 |
|
| 257 | + def test_non_regression_CDF_ISTP_with_proxy_and_config(self): |
| 258 | + ref = spz.get_data(spz.inventories.tree.amda.Parameters.MMS.MMS1.FPI.fast_mode.mms1_fpi_dismoms.mms1_dis_omni, |
| 259 | + "2021-06-01", "2021-06-08T02", output_format='CDF_ISTP') |
| 260 | + os.environ[amda_cfg.output_format.env_var_name] = 'CDF_ISTP' |
| 261 | + var = spz.get_data(spz.inventories.tree.amda.Parameters.MMS.MMS1.FPI.fast_mode.mms1_fpi_dismoms.mms1_dis_omni, |
| 262 | + "2021-06-01", "2021-06-08T02") |
| 263 | + self.assertTrue(len(ref.axes), 2) |
| 264 | + self.assertTrue(len(var.axes), 2) |
| 265 | + self.assertTrue(np.all(var.axes[1].values == ref.axes[1].values)) |
| 266 | + |
256 | 267 |
|
257 | 268 | if __name__ == '__main__':
|
258 | 269 | unittest.main()
|
0 commit comments