#!/usr/bin/env python
# coding: utf-8

# # Test: subset c3s-cmip6 HadGEM3-GC31-LL/ssp585

# In[ ]:


import os
os.environ['ROOK_URL'] = 'http://rook.dkrz.de/wps'

from rooki import operators as ops


# In[ ]:


wf = ops.Subset(
        ops.Input(
            'cmip6', ['c3s-cmip6.ScenarioMIP.MOHC.HadGEM3-GC31-LL.ssp585.r1i1p1f3.Amon.tas.gn.v20200114']
        ),
)
   


# In[ ]:


resp = wf.orchestrate()
assert resp.ok


# In[ ]:


resp.download_urls()


# In[ ]:


assert 'tas_Amon_HadGEM3-GC31-LL_ssp585_r1i1p1f3_gn_201501-204912.nc' in resp.download_urls()[0]


# In[ ]:


assert 'https://data.mips.copernicus-climate.eu/thredds/fileServer/esg_c3s-cmip6' in resp.download_urls()[0]


# In[ ]:


assert resp.num_files == 2, resp.num_files


# In[ ]:


assert int(resp.size_in_mb) == 55, resp.size_in_mb


# Provenance: 
# * expected duration=20 secs

# In[ ]:


from IPython.display import Image
Image(resp.provenance_image())