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

# # Test: subset HadGEM3-GC31-LL/ssp245
# 
# * Project: c3s-cmip6
# * Variable: tas
# * Frequency: day
# * *Complete* files

# 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.ssp245.r1i1p1f3.day.tas.gn.v20190908']
        ),
)
   


# In[ ]:


resp = wf.orchestrate()
assert resp.ok


# In[ ]:


resp.download_urls()


# In[ ]:


assert 'tas_day_HadGEM3-GC31-LL_ssp245_r1i1p1f3_gn_20150101-20491230.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) == 1684, resp.size_in_mb


# Provenance: 
# * expected duration=20 secs

# In[ ]:


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