Files
OpenBB/tests/gamestonk_terminal/custom/test_custom_model.py
jmaslek 6a66f3f3ed Custom data context (#1193)
* Add first iteration of custom context

* Add sample data + improve plot

* Change `head` to `show` with sorting and limit.  Add "-x" to plot and dynamic update of completer

* generate random time series for test csv

* Make columns lower case.  Check if date is in columns and convert to timestamp.  Improve plotting for dates

* Add qa to custom

* Add pred to custom

* Hugooooo

* Testing

* dang whitespace

Co-authored-by: Colin Delahunty <72827203+colin99d@users.noreply.github.com>
Co-authored-by: didierlopes.eth <dro.lopes@campus.fct.unl.pt>
2022-01-19 10:45:56 +00:00

25 lines
495 B
Python

import os
import pytest
from gamestonk_terminal.custom import custom_model
@pytest.mark.vcr()
@pytest.mark.parametrize(
"file",
[
"test.csv",
],
)
def test_test_load(recorder, file):
result_df = custom_model.load(os.path.join("custom_imports", file))
assert not result_df.empty
recorder.capture(result_df)
@pytest.mark.vcr()
def test_test_empty_load(recorder):
result_df = custom_model.load("")
assert result_df.empty
recorder.capture(result_df)