From 0e00aa103ffb026cabd16d35f9bfb942c8218928 Mon Sep 17 00:00:00 2001 From: Jean-Louis Fuchs Date: Thu, 12 Dec 2013 19:13:12 +0100 Subject: [PATCH] * created basic test --- test/test_get_code.py | 36 ++++++++++++++++++++++++++++++++++++ 1 file changed, 36 insertions(+) create mode 100644 test/test_get_code.py diff --git a/test/test_get_code.py b/test/test_get_code.py new file mode 100644 index 00000000..58932e6e --- /dev/null +++ b/test/test_get_code.py @@ -0,0 +1,36 @@ +import jedi.parser as parser +import difflib + +code_basic_features = ''' +def a_function(a_argument, a_default = "default"): + """A docstring""" + + a_result = 3 * a_argument + print(a_result) # a comment + if a_default == "default": + return str(a_result) + else + return None +''' + + +def diff_code_assert(a, b, n=4): + if a != b: + diff = "\n".join(difflib.unified_diff( + a.splitlines(), + b.splitlines(), + n=n, + lineterm="" + )) + assert False, "Code does not match:\n%s" % diff + pass + + +def test_basic_parsing(): + """Validate the parsing features""" + + prs = parser.Parser(code_basic_features) + diff_code_assert( + code_basic_features, + prs.top_module.get_code() + )