#!/usr/bin/python3 """test_tookkit.py - tests of the toolkit-module of DHParser Author: Eckhart Arnold Copyright 2017 Bavarian Academy of Sciences and Humanities Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. """ import os import sys from DHParser.toolkit import load_if_file class TestToolkit: filename = "tmp/test.py" if os.path.isdir('tmp') else "test/tmp/test.py" code1 = "x = 46" code2 = "def f():\n return 46" def setup(self): with open(self.filename, 'w') as f: f.write(self.code2) def teardown(self): os.remove(self.filename) def test_load_if_file(self): # an error should be raised if file expected but not found error_raised = False try: load_if_file('this_is_code_and_not_a_file') except FileNotFoundError: error_raised = True assert error_raised # multiline text will never be mistaken for a file assert load_if_file('this_is_code_and_not_a_file\n') # neither will text that does not look like a file name s = "this is code and not a file" assert s == load_if_file(s) # not a file and not mistaken for a file assert self.code1 == load_if_file(self.code1) # not a file and not mistaken for a file either assert self.code2 == load_if_file(self.code2) # file correctly loaded assert self.code2 == load_if_file(self.filename) if __name__ == "__main__": from run import runner runner("", globals())