Add some tests

Author: amorison

tests/test_cli.py

@@ -1,3 +1,4 @@
+import re
 import pytest
 import subprocess
 
@@ -62,3 +63,9 @@ def test_rprof_cli(all_cmd_rprof, tmpdir):
 
 def test_time_cli(all_cmd_time, tmpdir):
     helper_test_cli(all_cmd_time, tmpdir)
+
+
+def test_err_cli():
+    subp = subprocess.run('stagpy field', shell=True, capture_output=True)
+    reg = re.compile(br'^Oops!.*\nPlease.*\n\nNoParFileError.*$')
+    assert reg.match(subp.stderr)

tests/test_field.py

@@ -1,7 +1,21 @@
+import pytest
+import stagpy.error
 import stagpy.field
 import stagpy.phyvars
 
 
+def test_field_unknow(step):
+    with pytest.raises(stagpy.error.UnknownFieldVarError):
+        step.fields['InvalidField']
+    assert 'InvalidField' not in step.fields
+
+
+def test_field_missing(step):
+    with pytest.raises(stagpy.error.MissingDataError):
+        step.fields['rsc']
+    assert 'rsc' not in step.fields
+
+
 def test_valid_field_var():
     for var in stagpy.phyvars.FIELD:
         assert stagpy.field.valid_field_var(var)

tests/test_rprof.py

@@ -1,7 +1,24 @@
+import pytest
+import stagpy.error
 import stagpy.rprof
 import stagpy.phyvars
 
 
+def test_no_rprof_data(sdat):
+    with pytest.raises(stagpy.error.MissingDataError):
+        sdat.steps[1].rprofs['Tmean']
+
+
+def test_invalid_rprof(step):
+    with pytest.raises(stagpy.error.UnknownRprofVarError):
+        step.rprofs['DummyVar']
+
+
+def test_rprof_bounds_if_no_rprofs(sdat):
+    rcmb, rtot = sdat.steps[1].rprofs.bounds
+    assert rtot > rcmb
+
+
 def test_get_rprof(step):
     prof, rad, meta = step.rprofs['Tmean']
     assert rad is step.rprofs.centers

tests/test_stagyydata.py

@@ -93,5 +93,8 @@ def test_geom_refs(step):
     assert step.geom is step.fields.geom
 
 
-def test_geom_twod(step):
+def test_geom(step):
     assert step.geom.twod
+    assert not step.geom.threed
+    assert not step.geom.yinyang
+    assert step.geom.cartesian is not step.geom.spherical