15 lines
344 B
Python
15 lines
344 B
Python
import json
|
|
|
|
|
|
class MockData:
|
|
|
|
def __init__(self, data_dir) -> None:
|
|
self._data_dir = data_dir
|
|
|
|
def get_html(self, key: str) -> str:
|
|
return (self._data_dir / f"{key}.html").read_text()
|
|
|
|
def get_json(self, key: str) -> dict:
|
|
data = json.loads((self._data_dir / f"{key}.json").read_text())
|
|
return data
|