Преглед на файлове

refactor: pull out _read_json_from_file()

facilitates test mocking
main
Rob Hallam преди 2 месеца
родител
ревизия
778483837c
променени са 1 файла, в които са добавени 8 реда и са изтрити 0 реда
  1. +8
    -0
      pipeline/feature_extractors.py

+ 8
- 0
pipeline/feature_extractors.py Целия файл

@@ -356,6 +356,14 @@ class JSONFeatureExtractor(FeatureExtractor):
def _source_from_dict(self, d):
return Source(d["source"], d["path"], d["provider"])

def _read_json_from_file(self, file):
"""Read a JSON file and return the contents

Method exists to allow for mocking in tests
"""
with open(file, "r") as f:
return json.load(f)

def run(self):

# only pipeline JSON format for now


Зареждане…
Отказ
Запис