test_pangeo-notebook.py

import pytest
import importlib
import sys
import os

packages = [
    # these are problem libraries that don't always seem to import, mostly due
    # to dependencies outside the python world
    'netCDF4', 'h5py', 'rasterio', 'geopandas', 'cartopy', 'geoviews',
    # key packages
    'gcsfs', 's3fs', 'xarray', 'intake'
    ]

@pytest.mark.parametrize('package_name', packages, ids=packages)
def test_import(package_name):
    importlib.import_module(package_name)
    
def test_cartopy_downloads():
    import cartopy
    _ = cartopy.io.shapereader.natural_earth()  # should trigger download

def test_start():
    print(os.environ)
    if os.environ.get('PANGEO_ENV') is not None:
        assert os.environ['PANGEO_ENV'] == 'pangeo-notebook'