diff --git a/src/rpmquery/querylayeriniso.py b/src/rpmquery/querylayeriniso.py index c887fd7cb350592962c78423b0c2feb3c4223109..cfd34f9909767e08568d20f45424bc1342b385c3 100644 --- a/src/rpmquery/querylayeriniso.py +++ b/src/rpmquery/querylayeriniso.py @@ -12,17 +12,12 @@ # See the Mulan PSL v2 for more details. # ********************************************************************************** """ -import os -import sys -current_path = os.path.dirname(__file__) -parent_path = os.path.dirname(current_path) -sys.path.append(parent_path) -from rpmquery import RpmQuery -from utils.util import ISOUtils -from utils.depparse import ISODepParse -from utils.config import BaseConfig -from main.alglayer import AlgLayer +from .rpmquery import RpmQuery +from src.utils.util import ISOUtils +from src.utils.depparse import ISODepParse +from src.utils.config import BaseConfig +from src.main.alglayer import AlgLayer class QueryLayerInIso(RpmQuery): diff --git a/src/rpmquery/rpmquery.py b/src/rpmquery/rpmquery.py index 22e4bc4e08b9833915ac60610e34201490a7f222..fb76e64fc7684bbdc91432885aeddc62df14567e 100644 --- a/src/rpmquery/rpmquery.py +++ b/src/rpmquery/rpmquery.py @@ -15,7 +15,7 @@ import rpm -from rpmhdrinfo import RpmHdrInfo +from .rpmhdrinfo import RpmHdrInfo class RpmQuery(object): """ diff --git a/test/test_rpmcheck.py b/test/test_rpmcheck.py index 93707ce2bcf84ecec9e7c658de6df340ffee4c4a..7435802002be22ba6a745822ab68bbadefa5f3dc 100644 --- a/test/test_rpmcheck.py +++ b/test/test_rpmcheck.py @@ -37,6 +37,15 @@ class TestRpmCheck(unittest.TestCase): """ result = self.rpm_check.check_exist() self.assertIn(result, [True, False], "check_exist test failed!") + + def test_check_formate(self): + """ + Test class method check_formate() + Returns: + bool + """ + result = self.rpm_check.check_formate() + self.assertIn(result, [True, False], "check_formate test failed!") if __name__ == "__main__":