feat: add only_include_project report config flag

eth-brownie · Mar 28, 2021 · efc74f6 · efc74f6
1 parent f4fa2f8
commit efc74f6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/brownie/data/default-config.yaml b/brownie/data/default-config.yaml
@@ -50,6 +50,7 @@ console:
 reports:
     exclude_paths: null
     exclude_contracts: null
+    only_include_project: true
 
 hypothesis:
     deadline: null

diff --git a/brownie/test/output.py b/brownie/test/output.py
@@ -74,6 +74,7 @@ def _build_gas_profile_output():
 
     lines = [""]
 
+    only_include_project = CONFIG.settings["reports"]["only_include_project"]
     for full_name, values in sorted_gas:
         contract, function = full_name.split(".", 1)
 
@@ -82,7 +83,8 @@ def _build_gas_profile_output():
                 continue
         except (AttributeError, KeyError):
             # filters contracts that are not part of the project
-            continue
+            if only_include_project:
+                continue
         if contract in exclude_contracts:
             continue