Merge pull request #19 from mrstu/feature/test_calc_grid

Add lat lon fixtures and calc_grid shape test

Author: Joe Hamman

tests/test_vic2netcdf.py

@@ -1,17 +1,26 @@
 #!/usr/local/env python
-"""
-vic2netcdf_unit_test.py
-
-Set to run with pytest
+"""Set to run with pytest
 
-Usage: py.test (from RVIC or test directory)
+Usage: py.test
 """
+import pytest
+
 import numpy as np
 from tonic.tonic import calc_grid
 
 
-def test_calc_grid_standard():
-    lons = np.arange(0, 5, 0.25)
-    lats = np.arange(0, 10, 0.5)
+@pytest.fixture(scope="function")
+def lons():
+    return np.arange(0, 5, 0.25)
+
+    
+@pytest.fixture(scope="function")
+def lats():
+    return np.arange(0, 10, 0.5)
+
+
+def test_calc_grid_standard(lons, lats):
+    shape = np.meshgrid(lons,lats)[0].shape
     target_grid = calc_grid(lons, lats, decimals=4)
     assert type(target_grid) == dict
+    assert target_grid['mask'].shape == shape