fix structure test

Author: ArtemSmeta

tests/PiperOperator.py

@@ -15,23 +15,23 @@
 HEADERS = {"Content-Type": "application/json"}
 NER_RESPONSE_KEY = 'body'
 
+
 class PiperOperatorException(BaseException):
     def __init__(self, msg):
         pass
         # logger.exception(msg)
-    
 
 
 class FileLoadException(PiperOperatorException):
     def __init__(self, fn):
         self.fn = fn
-        super().__init__(f'file {fn} can`t be loaded')    
+        super().__init__(f'file {fn} can`t be loaded')
 
 
 class JSONGetKeyException(PiperOperatorException):
     def __init__(self, key):
         self.key = key
-        super().__init__(f'can`t get JSON key {key}')    
+        super().__init__(f'can`t get JSON key {key}')
 
 
 class NoAvailableModelsException(PiperOperatorException):
@@ -46,7 +46,8 @@ def get_data_by_key_from_response(cur_response, k):
     v = j.get(k)
     return v
 
-def get_data_by_key_from_url(url, key, post=True, data=None, file_name=""):    
+
+def get_data_by_key_from_url(url, key, post=True, data=None, file_name=""):
     try:
         if post:
             if file_name:
@@ -68,7 +69,7 @@ def get_data_by_key_from_url(url, key, post=True, data=None, file_name=""):
                 return val
             else:
                 return cur_response
-            
+
         else:
             cur_response = requests.get(url, headers=HEADERS, data=data)
             cur_response.raise_for_status()
@@ -83,7 +84,7 @@ def get_data_by_key_from_url(url, key, post=True, data=None, file_name=""):
         logger.exception(f'can`t get key from response: {cjke}')
 
     except Exception as e:
-        logger.exception(f'error while processing url {url}: {e}') 
+        logger.exception(f'error while processing url {url}: {e}')
 
 
 class PiperNLPWorker():
@@ -106,15 +107,14 @@ def __init__(self, base_url):
         # get named entitys from text url
         self.url_spacy_get_NE = f'{self.base_url}/extract_named_ents'
 
-
     def get_available_ner_models(self):
         return get_data_by_key_from_url(self.url_spacy_all_models, 'available_models', post=False)
 
     def set_current_spacy_model(self, model):
-        return get_data_by_key_from_url(self.url_spacy_set_model, '', post=True, data=json.dumps({'model_name':model}))
+        return get_data_by_key_from_url(self.url_spacy_set_model, '', post=True, data=json.dumps({'model_name': model}))
 
     def get_named_ent_from_text(self, txt):
-        resp = get_data_by_key_from_url(self.url_spacy_get_NE, 'result', post=False, data=json.dumps({'txt':txt}))
+        resp = get_data_by_key_from_url(self.url_spacy_get_NE, 'result', post=False, data=json.dumps({'txt': txt}))
         logger.debug(f'url is {resp}, response is {resp}')
         if NER_RESPONSE_KEY in resp.keys():
             named_ents = resp.get(NER_RESPONSE_KEY)
@@ -138,9 +138,9 @@ def get_text_from_file(self, fn):
     def set_tesseract_config(self, conf):
         return get_data_by_key_from_url(self.url_tsrct_cfg, '', post=True, data=json.dumps(conf))
 
+
 if __name__ == '__main__':
     piper_worker = PiperNLPWorker('http://localhost:8788')
-    
 
     amodels = piper_worker.get_available_ner_models()
     print('all models', amodels)
@@ -162,16 +162,12 @@ def set_tesseract_config(self, conf):
         logger.exception(e)
     # pprint(resp)
 
-
     txt = piper_worker.get_text_from_file('/home/pavel/repo/piper_new/piper/tests/ocr_data.pdf')
     logger.info(f'txt {txt}')
 
-
     ts_conf = dict()
     ts_conf['ts_lang'] = 'eng'
     ts_conf['ts_config_row'] = rf'--oem 1 --psm 6'
 
     resp = piper_worker.set_tesseract_config(ts_conf)
     logger.info(resp)
-    
-    

tests/__init__.py

@@ -1,14 +1,18 @@
 # pytest -vs tests/running_piper_test.py::TestDocker
 import requests
 
+from piper.envs import VirtualEnv
+
 main_app_url = f'http://localhost:8788'
 
-class TestDocker():
-    '''
+
+class TestDocker:
+    """
         Docker container API test. Methods:
             health_check
             run
-    '''
+    """
+
     def test_health_check(self):
         url = f'{main_app_url}/health_check'
         print(url)
@@ -29,10 +33,13 @@ def test_run(self):
         assert need_result == result.get('value')
 
 
-class TestVenv():
-    '''
+class TestVenv:
+    """
         venv container API test. Methods:
             dummy
-    '''
-    def test_dummy(self):
-        assert 1 / 0
+    """
+
+    with VirtualEnv() as env:
+        env.copy_struct_project()
+        env.create_files_for_venv()
+        env.create_files_for_tests()

tests/running_piper_test.py

@@ -21,19 +21,20 @@
 main_app_url = f'http://localhost:8788'
 file_path = Path(__file__).parent
 
+
 # pytest -vs tests/tsrct_test.py::TestTesseract::test_recognizer
-class TestTesseract():
-    '''
+class TestTesseract:
+    """
         Docker container API test. Methods:
             test_recognizer_jpg
             test_recognizer_pdf
             health_check
-    '''
+    """
 
     def test_recognizer_jpg(self):
-        '''
+        """
             jpg file recognize test
-        '''
+        """
         fn = file_path.joinpath('ocr_data.jpg')
         url = f'{main_app_url}/recognize'
 
@@ -49,29 +50,27 @@ def test_recognizer_jpg(self):
         except Exception as e:
             pytest.raises(Exception)
 
-
     def test_recognizer_pdf(self):
-        '''
+        """
             pdf file recognize test
-        '''
+        """
         fn = file_path.joinpath('ocr_data.pdf')
         url = f'{main_app_url}/recognize'
 
         result = tu.send_file_to_service(url, fn)
         print(result.status_code)
-        assert result.status_code == 200        
+        assert result.status_code == 200
         try:
             data = result.json()
             print('data', data)
             assert len(data) != 0
         except Exception as e:
             pytest.raises(Exception)
 
-
     def test_health_check(self):
-        '''
+        """
             health check test
-        '''
+        """
         url = f'{main_app_url}/health_check'
         print(url)
         result = requests.post(url)

tests/tsrct_test.py

@@ -30,7 +30,6 @@
 SOURCE_FOLDER = file_path
 OUTPUT_FOLDER = file_path.joinpath('out')
 
-
 if __name__ == '__main__':
     cfg = get_configuration()
 
@@ -43,7 +42,7 @@
     ts_conf = dict()
     ts_conf['ts_lang'] = 'eng'
 
-    for oem in cfg.available_OEM: 
+    for oem in cfg.available_OEM:
         for psm in cfg.available_PSM:
             # change tesseract config
             ts_conf['ts_config_row'] = rf'--oem {oem} --psm {psm}'