Merge pull request #126 from fireeye/fix-testbed

Fix testbed

scripts/testbed/run_rule_on_testbed.py

@@ -18,9 +18,12 @@ import argparse
 import capa.main
 import capa.rules
 import capa.features.freeze
-from scripts.testbed import FNAMES_EXTENSION, FREEZE_EXTENSION
+from utils import FNAMES_EXTENSION, FREEZE_EXTENSION
 from start_ida_export_fimages import export_fimages
 
+# enable import of code in the same directory
+sys.path.append(os.path.dirname(__file__))
+
 logger = logging.getLogger(__name__)
 
 # sorry globals...

scripts/testbed/start_ida_dump_fnames.py

@@ -17,7 +17,10 @@ import subprocess
 
 import argparse
 
-from scripts.testbed import FNAMES_EXTENSION
+from utils import FNAMES_EXTENSION
+
+# enable import of code in the same directory
+sys.path.append(os.path.dirname(__file__))
 
 IDA32_PATH = "C:\\Program Files\\IDA Pro 7.3\\ida.exe"
 IDA64_PATH = "C:\\Program Files\\IDA Pro 7.3\\ida64.exe"