Update tests

andykee · Jul 8, 2020 · 1c46428 · 1c46428
1 parent 3a98460
commit 1c46428
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/tests/test_cache.py b/tests/test_cache.py
@@ -17,6 +17,11 @@ def test_cache_delete():
     assert 'value' not in cache.keys()
 
 
+def test_cache_delete_no_key():
+    cache = lentil.cache.Cache()
+    cache.delete('value')
+
+
 def test_cache_clear():
     cache = lentil.cache.Cache()
     value = np.random.random()

diff --git a/tests/test_plane.py b/tests/test_plane.py
@@ -26,6 +26,7 @@ def test_default_plane():
     assert p.mask == p.amplitude
     assert p.segmask is None
     assert p.nseg == 0
+    assert p.cache_attrs == ['amplitude', 'phase']
 
 
 def test_set_plane_attrs_none():