Add new test test_rdoc_to_capa

This commit is contained in:
Pratham Chauhan
2023-03-29 22:57:11 +05:30
parent 237554d84a
commit c3fdab8ec5

View File

@@ -232,4 +232,9 @@ def test_basic_block_node_from_capa():
def test_json_to_rdoc(): def test_json_to_rdoc():
path = fixtures.get_data_path_by_name("pma01-01-rd") path = fixtures.get_data_path_by_name("pma01-01-rd")
assert isinstance(rdoc.ResultDocument.parse_file(path),rdoc.ResultDocument) assert isinstance(rdoc.ResultDocument.parse_file(path),rdoc.ResultDocument)
def test_rdoc_to_capa():
path = fixtures.get_data_path_by_name("pma01-01-rd")
assert len(rdoc.ResultDocument.parse_file(path).to_capa()) ==2
assert isinstance(rdoc.ResultDocument.parse_file(path).to_capa(),tuple)