Test ParsedCommand.evaluate_test

glue-viz · May 3, 2016 · 0d78c79 · 0d78c79
1 parent c7b17dd
commit 0d78c79
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/glue/core/tests/test_parse.py b/glue/core/tests/test_parse.py
@@ -133,6 +133,22 @@ def test_evaluate_math(self):
         pc = parse.ParsedCommand(cmd, refs)
         assert pc.evaluate(data) == 1
 
+    def test_evaluate_test(self):
+
+        data = MagicMock()
+        c1 = ComponentID('c1')
+        data.__getitem__.return_value = 10
+        refs = {'comp1': c1}
+
+        cmd = 'numpy.log10({comp1}) + 3.4 - {comp1}'
+        pc = parse.ParsedCommand(cmd, refs)
+        pc.evaluate_test()
+
+        cmd = 'nump.log10({comp1}) + 3.4 - {comp1}'
+        pc = parse.ParsedCommand(cmd, refs)
+        with pytest.raises(NameError) as exc:
+            pc.evaluate_test()
+        assert exc.value.args[0] == "name 'nump' is not defined"
 
 
 class TestParsedComponentLink(object):