FilesExpand file tree

 master

/

test_preprocessor_intermediate.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

29 lines (20 loc) · 1.03 KB

 master

/

test_preprocessor_intermediate.py

Top

File metadata and controls

• Code
• Blame

29 lines (20 loc) · 1.03 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

import os

import unittest

import re

from shellpython import preprocessor

class TestPreprocessorIntermediate(unittest.TestCase):

def setUp(self):

cur_dir = os.path.split(__file__)[0]

self.test_dir = os.path.join(cur_dir, 'data', 'preprocessor')

def test_process_and_compare_file(self):

with open(os.path.join(self.test_dir, 'test.spy')) as code_f:

imdt = preprocessor._preprocess_code_to_intermediate(code_f.read())

with open(os.path.join(self.test_dir, 'test.imdt')) as imdt_etalon_f:

imdt_etalon = imdt_etalon_f.read()

statement_regex = re.compile(r'((.+\n)+)')

imdt_matches = [[match[0]] for match in statement_regex.findall(imdt)]

imdt_etalon_matches = [[match[0]] for match in statement_regex.findall(imdt_etalon)]

self.assertEqual(len(imdt_matches), len(imdt_etalon_matches))

zipped_matches = map(list.__add__, imdt_matches, imdt_etalon_matches)

for matches in zipped_matches:

self.assertEqual(matches[0], matches[1])