mirror of
https://github.com/freqtrade/freqtrade.git
synced 2024-11-10 02:12:01 +00:00
ruff format: Update tests/ base directory
This commit is contained in:
Matthias
parent
53947732a0
commit
23427bec08
5410
tests/conftest.py
5410
tests/conftest.py
File diff suppressed because it is too large
Load Diff
|
|
@ -12,7 +12,7 @@ from tests.conftest import CURRENT_TEST_STRATEGY
|
|||
|
||||
# Parse common command-line-arguments. Used for all tools
|
||||
def test_parse_args_none() -> None:
|
||||
arguments = Arguments(['trade'])
|
||||
arguments = Arguments(["trade"])
|
||||
assert isinstance(arguments, Arguments)
|
||||
x = arguments.get_parsed_arg()
|
||||
assert isinstance(x, dict)
|
||||
|
|
@ -20,265 +20,278 @@ def test_parse_args_none() -> None:
|
|||
|
||||
|
||||
def test_parse_args_defaults(mocker) -> None:
|
||||
mocker.patch.object(Path, 'is_file', MagicMock(side_effect=[False, True]))
|
||||
args = Arguments(['trade']).get_parsed_arg()
|
||||
assert args['config'] == ['config.json']
|
||||
assert args['strategy_path'] is None
|
||||
assert args['datadir'] is None
|
||||
assert args['verbosity'] == 0
|
||||
mocker.patch.object(Path, "is_file", MagicMock(side_effect=[False, True]))
|
||||
args = Arguments(["trade"]).get_parsed_arg()
|
||||
assert args["config"] == ["config.json"]
|
||||
assert args["strategy_path"] is None
|
||||
assert args["datadir"] is None
|
||||
assert args["verbosity"] == 0
|
||||
|
||||
|
||||
def test_parse_args_default_userdatadir(mocker) -> None:
|
||||
mocker.patch.object(Path, 'is_file', MagicMock(return_value=True))
|
||||
args = Arguments(['trade']).get_parsed_arg()
|
||||
mocker.patch.object(Path, "is_file", MagicMock(return_value=True))
|
||||
args = Arguments(["trade"]).get_parsed_arg()
|
||||
# configuration defaults to user_data if that is available.
|
||||
assert args['config'] == [str(Path('user_data/config.json'))]
|
||||
assert args['strategy_path'] is None
|
||||
assert args['datadir'] is None
|
||||
assert args['verbosity'] == 0
|
||||
assert args["config"] == [str(Path("user_data/config.json"))]
|
||||
assert args["strategy_path"] is None
|
||||
assert args["datadir"] is None
|
||||
assert args["verbosity"] == 0
|
||||
|
||||
|
||||
def test_parse_args_userdatadir(mocker) -> None:
|
||||
mocker.patch.object(Path, 'is_file', MagicMock(return_value=True))
|
||||
args = Arguments(['trade', '--user-data-dir', 'user_data']).get_parsed_arg()
|
||||
mocker.patch.object(Path, "is_file", MagicMock(return_value=True))
|
||||
args = Arguments(["trade", "--user-data-dir", "user_data"]).get_parsed_arg()
|
||||
# configuration defaults to user_data if that is available.
|
||||
assert args['config'] == [str(Path('user_data/config.json'))]
|
||||
assert args['strategy_path'] is None
|
||||
assert args['datadir'] is None
|
||||
assert args['verbosity'] == 0
|
||||
assert args["config"] == [str(Path("user_data/config.json"))]
|
||||
assert args["strategy_path"] is None
|
||||
assert args["datadir"] is None
|
||||
assert args["verbosity"] == 0
|
||||
|
||||
|
||||
def test_parse_args_config() -> None:
|
||||
args = Arguments(['trade', '-c', '/dev/null']).get_parsed_arg()
|
||||
assert args['config'] == ['/dev/null']
|
||||
args = Arguments(["trade", "-c", "/dev/null"]).get_parsed_arg()
|
||||
assert args["config"] == ["/dev/null"]
|
||||
|
||||
args = Arguments(['trade', '--config', '/dev/null']).get_parsed_arg()
|
||||
assert args['config'] == ['/dev/null']
|
||||
args = Arguments(["trade", "--config", "/dev/null"]).get_parsed_arg()
|
||||
assert args["config"] == ["/dev/null"]
|
||||
|
||||
args = Arguments(['trade', '--config', '/dev/null',
|
||||
'--config', '/dev/zero'],).get_parsed_arg()
|
||||
assert args['config'] == ['/dev/null', '/dev/zero']
|
||||
args = Arguments(
|
||||
["trade", "--config", "/dev/null", "--config", "/dev/zero"],
|
||||
).get_parsed_arg()
|
||||
assert args["config"] == ["/dev/null", "/dev/zero"]
|
||||
|
||||
|
||||
def test_parse_args_db_url() -> None:
|
||||
args = Arguments(['trade', '--db-url', 'sqlite:///test.sqlite']).get_parsed_arg()
|
||||
assert args['db_url'] == 'sqlite:///test.sqlite'
|
||||
args = Arguments(["trade", "--db-url", "sqlite:///test.sqlite"]).get_parsed_arg()
|
||||
assert args["db_url"] == "sqlite:///test.sqlite"
|
||||
|
||||
|
||||
def test_parse_args_verbose() -> None:
|
||||
args = Arguments(['trade', '-v']).get_parsed_arg()
|
||||
assert args['verbosity'] == 1
|
||||
args = Arguments(["trade", "-v"]).get_parsed_arg()
|
||||
assert args["verbosity"] == 1
|
||||
|
||||
args = Arguments(['trade', '--verbose']).get_parsed_arg()
|
||||
assert args['verbosity'] == 1
|
||||
args = Arguments(["trade", "--verbose"]).get_parsed_arg()
|
||||
assert args["verbosity"] == 1
|
||||
|
||||
|
||||
def test_common_scripts_options() -> None:
|
||||
args = Arguments(['download-data', '-p', 'ETH/BTC', 'XRP/BTC']).get_parsed_arg()
|
||||
args = Arguments(["download-data", "-p", "ETH/BTC", "XRP/BTC"]).get_parsed_arg()
|
||||
|
||||
assert args['pairs'] == ['ETH/BTC', 'XRP/BTC']
|
||||
assert 'func' in args
|
||||
assert args["pairs"] == ["ETH/BTC", "XRP/BTC"]
|
||||
assert "func" in args
|
||||
|
||||
|
||||
def test_parse_args_version() -> None:
|
||||
with pytest.raises(SystemExit, match=r'0'):
|
||||
Arguments(['--version']).get_parsed_arg()
|
||||
with pytest.raises(SystemExit, match=r"0"):
|
||||
Arguments(["--version"]).get_parsed_arg()
|
||||
|
||||
|
||||
def test_parse_args_invalid() -> None:
|
||||
with pytest.raises(SystemExit, match=r'2'):
|
||||
Arguments(['-c']).get_parsed_arg()
|
||||
with pytest.raises(SystemExit, match=r"2"):
|
||||
Arguments(["-c"]).get_parsed_arg()
|
||||
|
||||
|
||||
def test_parse_args_strategy() -> None:
|
||||
args = Arguments(['trade', '--strategy', 'SomeStrategy']).get_parsed_arg()
|
||||
assert args['strategy'] == 'SomeStrategy'
|
||||
args = Arguments(["trade", "--strategy", "SomeStrategy"]).get_parsed_arg()
|
||||
assert args["strategy"] == "SomeStrategy"
|
||||
|
||||
|
||||
def test_parse_args_strategy_invalid() -> None:
|
||||
with pytest.raises(SystemExit, match=r'2'):
|
||||
Arguments(['--strategy']).get_parsed_arg()
|
||||
with pytest.raises(SystemExit, match=r"2"):
|
||||
Arguments(["--strategy"]).get_parsed_arg()
|
||||
|
||||
|
||||
def test_parse_args_strategy_path() -> None:
|
||||
args = Arguments(['trade', '--strategy-path', '/some/path']).get_parsed_arg()
|
||||
assert args['strategy_path'] == '/some/path'
|
||||
args = Arguments(["trade", "--strategy-path", "/some/path"]).get_parsed_arg()
|
||||
assert args["strategy_path"] == "/some/path"
|
||||
|
||||
|
||||
def test_parse_args_strategy_path_invalid() -> None:
|
||||
with pytest.raises(SystemExit, match=r'2'):
|
||||
Arguments(['--strategy-path']).get_parsed_arg()
|
||||
with pytest.raises(SystemExit, match=r"2"):
|
||||
Arguments(["--strategy-path"]).get_parsed_arg()
|
||||
|
||||
|
||||
def test_parse_args_backtesting_invalid() -> None:
|
||||
with pytest.raises(SystemExit, match=r'2'):
|
||||
Arguments(['backtesting --timeframe']).get_parsed_arg()
|
||||
with pytest.raises(SystemExit, match=r"2"):
|
||||
Arguments(["backtesting --timeframe"]).get_parsed_arg()
|
||||
|
||||
with pytest.raises(SystemExit, match=r'2'):
|
||||
Arguments(['backtesting --timeframe', 'abc']).get_parsed_arg()
|
||||
with pytest.raises(SystemExit, match=r"2"):
|
||||
Arguments(["backtesting --timeframe", "abc"]).get_parsed_arg()
|
||||
|
||||
|
||||
def test_parse_args_backtesting_custom() -> None:
|
||||
args = [
|
||||
'backtesting',
|
||||
'-c', 'test_conf.json',
|
||||
'--timeframe', '1m',
|
||||
'--strategy-list',
|
||||
"backtesting",
|
||||
"-c",
|
||||
"test_conf.json",
|
||||
"--timeframe",
|
||||
"1m",
|
||||
"--strategy-list",
|
||||
CURRENT_TEST_STRATEGY,
|
||||
'SampleStrategy'
|
||||
"SampleStrategy",
|
||||
]
|
||||
call_args = Arguments(args).get_parsed_arg()
|
||||
assert call_args['config'] == ['test_conf.json']
|
||||
assert call_args['verbosity'] == 0
|
||||
assert call_args['command'] == 'backtesting'
|
||||
assert call_args['func'] is not None
|
||||
assert call_args['timeframe'] == '1m'
|
||||
assert isinstance(call_args['strategy_list'], list)
|
||||
assert len(call_args['strategy_list']) == 2
|
||||
assert call_args["config"] == ["test_conf.json"]
|
||||
assert call_args["verbosity"] == 0
|
||||
assert call_args["command"] == "backtesting"
|
||||
assert call_args["func"] is not None
|
||||
assert call_args["timeframe"] == "1m"
|
||||
assert isinstance(call_args["strategy_list"], list)
|
||||
assert len(call_args["strategy_list"]) == 2
|
||||
|
||||
|
||||
def test_parse_args_hyperopt_custom() -> None:
|
||||
args = [
|
||||
'hyperopt',
|
||||
'-c', 'test_conf.json',
|
||||
'--epochs', '20',
|
||||
'--spaces', 'buy'
|
||||
]
|
||||
args = ["hyperopt", "-c", "test_conf.json", "--epochs", "20", "--spaces", "buy"]
|
||||
call_args = Arguments(args).get_parsed_arg()
|
||||
assert call_args['config'] == ['test_conf.json']
|
||||
assert call_args['epochs'] == 20
|
||||
assert call_args['verbosity'] == 0
|
||||
assert call_args['command'] == 'hyperopt'
|
||||
assert call_args['spaces'] == ['buy']
|
||||
assert call_args['func'] is not None
|
||||
assert callable(call_args['func'])
|
||||
assert call_args["config"] == ["test_conf.json"]
|
||||
assert call_args["epochs"] == 20
|
||||
assert call_args["verbosity"] == 0
|
||||
assert call_args["command"] == "hyperopt"
|
||||
assert call_args["spaces"] == ["buy"]
|
||||
assert call_args["func"] is not None
|
||||
assert callable(call_args["func"])
|
||||
|
||||
|
||||
def test_download_data_options() -> None:
|
||||
args = [
|
||||
'download-data',
|
||||
'--datadir', 'datadir/directory',
|
||||
'--pairs-file', 'file_with_pairs',
|
||||
'--days', '30',
|
||||
'--exchange', 'binance'
|
||||
"download-data",
|
||||
"--datadir",
|
||||
"datadir/directory",
|
||||
"--pairs-file",
|
||||
"file_with_pairs",
|
||||
"--days",
|
||||
"30",
|
||||
"--exchange",
|
||||
"binance",
|
||||
]
|
||||
pargs = Arguments(args).get_parsed_arg()
|
||||
|
||||
assert pargs['pairs_file'] == 'file_with_pairs'
|
||||
assert pargs['datadir'] == 'datadir/directory'
|
||||
assert pargs['days'] == 30
|
||||
assert pargs['exchange'] == 'binance'
|
||||
assert pargs["pairs_file"] == "file_with_pairs"
|
||||
assert pargs["datadir"] == "datadir/directory"
|
||||
assert pargs["days"] == 30
|
||||
assert pargs["exchange"] == "binance"
|
||||
|
||||
|
||||
def test_plot_dataframe_options() -> None:
|
||||
args = [
|
||||
'plot-dataframe',
|
||||
'-c', 'tests/testdata/testconfigs/main_test_config.json',
|
||||
'--indicators1', 'sma10', 'sma100',
|
||||
'--indicators2', 'macd', 'fastd', 'fastk',
|
||||
'--plot-limit', '30',
|
||||
'-p', 'UNITTEST/BTC',
|
||||
"plot-dataframe",
|
||||
"-c",
|
||||
"tests/testdata/testconfigs/main_test_config.json",
|
||||
"--indicators1",
|
||||
"sma10",
|
||||
"sma100",
|
||||
"--indicators2",
|
||||
"macd",
|
||||
"fastd",
|
||||
"fastk",
|
||||
"--plot-limit",
|
||||
"30",
|
||||
"-p",
|
||||
"UNITTEST/BTC",
|
||||
]
|
||||
pargs = Arguments(args).get_parsed_arg()
|
||||
|
||||
assert pargs['indicators1'] == ['sma10', 'sma100']
|
||||
assert pargs['indicators2'] == ['macd', 'fastd', 'fastk']
|
||||
assert pargs['plot_limit'] == 30
|
||||
assert pargs['pairs'] == ['UNITTEST/BTC']
|
||||
assert pargs["indicators1"] == ["sma10", "sma100"]
|
||||
assert pargs["indicators2"] == ["macd", "fastd", "fastk"]
|
||||
assert pargs["plot_limit"] == 30
|
||||
assert pargs["pairs"] == ["UNITTEST/BTC"]
|
||||
|
||||
|
||||
@pytest.mark.parametrize('auto_open_arg', [True, False])
|
||||
@pytest.mark.parametrize("auto_open_arg", [True, False])
|
||||
def test_plot_profit_options(auto_open_arg: bool) -> None:
|
||||
args = [
|
||||
'plot-profit',
|
||||
'-p', 'UNITTEST/BTC',
|
||||
'--trade-source', 'DB',
|
||||
'--db-url', 'sqlite:///whatever.sqlite',
|
||||
"plot-profit",
|
||||
"-p",
|
||||
"UNITTEST/BTC",
|
||||
"--trade-source",
|
||||
"DB",
|
||||
"--db-url",
|
||||
"sqlite:///whatever.sqlite",
|
||||
]
|
||||
if auto_open_arg:
|
||||
args.append('--auto-open')
|
||||
args.append("--auto-open")
|
||||
pargs = Arguments(args).get_parsed_arg()
|
||||
|
||||
assert pargs['trade_source'] == 'DB'
|
||||
assert pargs['pairs'] == ['UNITTEST/BTC']
|
||||
assert pargs['db_url'] == 'sqlite:///whatever.sqlite'
|
||||
assert pargs['plot_auto_open'] == auto_open_arg
|
||||
assert pargs["trade_source"] == "DB"
|
||||
assert pargs["pairs"] == ["UNITTEST/BTC"]
|
||||
assert pargs["db_url"] == "sqlite:///whatever.sqlite"
|
||||
assert pargs["plot_auto_open"] == auto_open_arg
|
||||
|
||||
|
||||
def test_config_notallowed(mocker) -> None:
|
||||
mocker.patch.object(Path, 'is_file', MagicMock(return_value=False))
|
||||
mocker.patch.object(Path, "is_file", MagicMock(return_value=False))
|
||||
args = [
|
||||
'create-userdir',
|
||||
"create-userdir",
|
||||
]
|
||||
pargs = Arguments(args).get_parsed_arg()
|
||||
|
||||
assert 'config' not in pargs
|
||||
assert "config" not in pargs
|
||||
|
||||
# When file exists:
|
||||
mocker.patch.object(Path, 'is_file', MagicMock(return_value=True))
|
||||
mocker.patch.object(Path, "is_file", MagicMock(return_value=True))
|
||||
args = [
|
||||
'create-userdir',
|
||||
"create-userdir",
|
||||
]
|
||||
pargs = Arguments(args).get_parsed_arg()
|
||||
# config is not added even if it exists, since create-userdir is in the notallowed list
|
||||
assert 'config' not in pargs
|
||||
assert "config" not in pargs
|
||||
|
||||
|
||||
def test_config_notrequired(mocker) -> None:
|
||||
mocker.patch.object(Path, 'is_file', MagicMock(return_value=False))
|
||||
mocker.patch.object(Path, "is_file", MagicMock(return_value=False))
|
||||
args = [
|
||||
'download-data',
|
||||
"download-data",
|
||||
]
|
||||
pargs = Arguments(args).get_parsed_arg()
|
||||
|
||||
assert pargs['config'] is None
|
||||
assert pargs["config"] is None
|
||||
|
||||
# When file exists:
|
||||
mocker.patch.object(Path, 'is_file', MagicMock(side_effect=[False, True]))
|
||||
mocker.patch.object(Path, "is_file", MagicMock(side_effect=[False, True]))
|
||||
args = [
|
||||
'download-data',
|
||||
"download-data",
|
||||
]
|
||||
pargs = Arguments(args).get_parsed_arg()
|
||||
# config is added if it exists
|
||||
assert pargs['config'] == ['config.json']
|
||||
assert pargs["config"] == ["config.json"]
|
||||
|
||||
|
||||
def test_check_int_positive() -> None:
|
||||
assert check_int_positive('3') == 3
|
||||
assert check_int_positive('1') == 1
|
||||
assert check_int_positive('100') == 100
|
||||
assert check_int_positive("3") == 3
|
||||
assert check_int_positive("1") == 1
|
||||
assert check_int_positive("100") == 100
|
||||
|
||||
with pytest.raises(argparse.ArgumentTypeError):
|
||||
check_int_positive('-2')
|
||||
check_int_positive("-2")
|
||||
|
||||
with pytest.raises(argparse.ArgumentTypeError):
|
||||
check_int_positive('0')
|
||||
check_int_positive("0")
|
||||
|
||||
with pytest.raises(argparse.ArgumentTypeError):
|
||||
check_int_positive(0)
|
||||
|
||||
with pytest.raises(argparse.ArgumentTypeError):
|
||||
check_int_positive('3.5')
|
||||
check_int_positive("3.5")
|
||||
|
||||
with pytest.raises(argparse.ArgumentTypeError):
|
||||
check_int_positive('DeadBeef')
|
||||
check_int_positive("DeadBeef")
|
||||
|
||||
|
||||
def test_check_int_nonzero() -> None:
|
||||
assert check_int_nonzero('3') == 3
|
||||
assert check_int_nonzero('1') == 1
|
||||
assert check_int_nonzero('100') == 100
|
||||
assert check_int_nonzero("3") == 3
|
||||
assert check_int_nonzero("1") == 1
|
||||
assert check_int_nonzero("100") == 100
|
||||
|
||||
assert check_int_nonzero('-2') == -2
|
||||
assert check_int_nonzero("-2") == -2
|
||||
|
||||
with pytest.raises(argparse.ArgumentTypeError):
|
||||
check_int_nonzero('0')
|
||||
check_int_nonzero("0")
|
||||
|
||||
with pytest.raises(argparse.ArgumentTypeError):
|
||||
check_int_nonzero(0)
|
||||
|
||||
with pytest.raises(argparse.ArgumentTypeError):
|
||||
check_int_nonzero('3.5')
|
||||
check_int_nonzero("3.5")
|
||||
|
||||
with pytest.raises(argparse.ArgumentTypeError):
|
||||
check_int_nonzero('DeadBeef')
|
||||
check_int_nonzero("DeadBeef")
|
||||
|
|
|
|||
File diff suppressed because it is too large
Load Diff
|
|
@ -17,93 +17,100 @@ from tests.conftest import log_has, log_has_re
|
|||
|
||||
def test_create_datadir(mocker, default_conf, caplog) -> None:
|
||||
mocker.patch.object(Path, "is_dir", MagicMock(return_value=False))
|
||||
md = mocker.patch.object(Path, 'mkdir', MagicMock())
|
||||
md = mocker.patch.object(Path, "mkdir", MagicMock())
|
||||
|
||||
create_datadir(default_conf, '/foo/bar')
|
||||
assert md.call_args[1]['parents'] is True
|
||||
assert log_has('Created data directory: /foo/bar', caplog)
|
||||
create_datadir(default_conf, "/foo/bar")
|
||||
assert md.call_args[1]["parents"] is True
|
||||
assert log_has("Created data directory: /foo/bar", caplog)
|
||||
|
||||
|
||||
def test_create_userdata_dir(mocker, default_conf, caplog) -> None:
|
||||
mocker.patch.object(Path, "is_dir", MagicMock(return_value=False))
|
||||
md = mocker.patch.object(Path, 'mkdir', MagicMock())
|
||||
md = mocker.patch.object(Path, "mkdir", MagicMock())
|
||||
|
||||
x = create_userdata_dir('/tmp/bar', create_dir=True)
|
||||
x = create_userdata_dir("/tmp/bar", create_dir=True)
|
||||
assert md.call_count == 10
|
||||
assert md.call_args[1]['parents'] is False
|
||||
assert md.call_args[1]["parents"] is False
|
||||
assert log_has(f'Created user-data directory: {Path("/tmp/bar")}', caplog)
|
||||
assert isinstance(x, Path)
|
||||
assert str(x) == str(Path("/tmp/bar"))
|
||||
|
||||
|
||||
def test_create_userdata_dir_and_chown(mocker, tmp_path, caplog) -> None:
|
||||
sp_mock = mocker.patch('subprocess.check_output')
|
||||
path = tmp_path / 'bar'
|
||||
sp_mock = mocker.patch("subprocess.check_output")
|
||||
path = tmp_path / "bar"
|
||||
assert not path.is_dir()
|
||||
|
||||
x = create_userdata_dir(str(path), create_dir=True)
|
||||
assert sp_mock.call_count == 0
|
||||
assert log_has(f'Created user-data directory: {path}', caplog)
|
||||
assert log_has(f"Created user-data directory: {path}", caplog)
|
||||
assert isinstance(x, Path)
|
||||
assert path.is_dir()
|
||||
assert (path / 'data').is_dir()
|
||||
assert (path / "data").is_dir()
|
||||
|
||||
os.environ['FT_APP_ENV'] = 'docker'
|
||||
chown_user_directory(path / 'data')
|
||||
os.environ["FT_APP_ENV"] = "docker"
|
||||
chown_user_directory(path / "data")
|
||||
assert sp_mock.call_count == 1
|
||||
del os.environ['FT_APP_ENV']
|
||||
del os.environ["FT_APP_ENV"]
|
||||
|
||||
|
||||
def test_create_userdata_dir_exists(mocker, default_conf, caplog) -> None:
|
||||
mocker.patch.object(Path, "is_dir", MagicMock(return_value=True))
|
||||
md = mocker.patch.object(Path, 'mkdir', MagicMock())
|
||||
md = mocker.patch.object(Path, "mkdir", MagicMock())
|
||||
|
||||
create_userdata_dir('/tmp/bar')
|
||||
create_userdata_dir("/tmp/bar")
|
||||
assert md.call_count == 0
|
||||
|
||||
|
||||
def test_create_userdata_dir_exists_exception(mocker, default_conf, caplog) -> None:
|
||||
mocker.patch.object(Path, "is_dir", MagicMock(return_value=False))
|
||||
md = mocker.patch.object(Path, 'mkdir', MagicMock())
|
||||
md = mocker.patch.object(Path, "mkdir", MagicMock())
|
||||
|
||||
with pytest.raises(OperationalException,
|
||||
match=r'Directory `.{1,2}tmp.{1,2}bar` does not exist.*'):
|
||||
create_userdata_dir('/tmp/bar', create_dir=False)
|
||||
with pytest.raises(
|
||||
OperationalException, match=r"Directory `.{1,2}tmp.{1,2}bar` does not exist.*"
|
||||
):
|
||||
create_userdata_dir("/tmp/bar", create_dir=False)
|
||||
assert md.call_count == 0
|
||||
|
||||
|
||||
def test_copy_sample_files(mocker, default_conf, caplog) -> None:
|
||||
mocker.patch.object(Path, "is_dir", MagicMock(return_value=True))
|
||||
mocker.patch.object(Path, "exists", MagicMock(return_value=False))
|
||||
copymock = mocker.patch('shutil.copy', MagicMock())
|
||||
copymock = mocker.patch("shutil.copy", MagicMock())
|
||||
|
||||
copy_sample_files(Path('/tmp/bar'))
|
||||
copy_sample_files(Path("/tmp/bar"))
|
||||
assert copymock.call_count == 3
|
||||
assert copymock.call_args_list[0][0][1] == str(
|
||||
Path('/tmp/bar') / 'strategies/sample_strategy.py')
|
||||
Path("/tmp/bar") / "strategies/sample_strategy.py"
|
||||
)
|
||||
assert copymock.call_args_list[1][0][1] == str(
|
||||
Path('/tmp/bar') / 'hyperopts/sample_hyperopt_loss.py')
|
||||
Path("/tmp/bar") / "hyperopts/sample_hyperopt_loss.py"
|
||||
)
|
||||
assert copymock.call_args_list[2][0][1] == str(
|
||||
Path('/tmp/bar') / 'notebooks/strategy_analysis_example.ipynb')
|
||||
Path("/tmp/bar") / "notebooks/strategy_analysis_example.ipynb"
|
||||
)
|
||||
|
||||
|
||||
def test_copy_sample_files_errors(mocker, default_conf, caplog) -> None:
|
||||
mocker.patch.object(Path, "is_dir", MagicMock(return_value=False))
|
||||
mocker.patch.object(Path, "exists", MagicMock(return_value=False))
|
||||
mocker.patch('shutil.copy', MagicMock())
|
||||
with pytest.raises(OperationalException,
|
||||
match=r"Directory `.{1,2}tmp.{1,2}bar` does not exist\."):
|
||||
copy_sample_files(Path('/tmp/bar'))
|
||||
mocker.patch("shutil.copy", MagicMock())
|
||||
with pytest.raises(
|
||||
OperationalException, match=r"Directory `.{1,2}tmp.{1,2}bar` does not exist\."
|
||||
):
|
||||
copy_sample_files(Path("/tmp/bar"))
|
||||
|
||||
mocker.patch.object(Path, "is_dir", MagicMock(side_effect=[True, False]))
|
||||
|
||||
with pytest.raises(OperationalException,
|
||||
match=r"Directory `.{1,2}tmp.{1,2}bar.{1,2}strategies` does not exist\."):
|
||||
copy_sample_files(Path('/tmp/bar'))
|
||||
with pytest.raises(
|
||||
OperationalException,
|
||||
match=r"Directory `.{1,2}tmp.{1,2}bar.{1,2}strategies` does not exist\.",
|
||||
):
|
||||
copy_sample_files(Path("/tmp/bar"))
|
||||
mocker.patch.object(Path, "is_dir", MagicMock(return_value=True))
|
||||
mocker.patch.object(Path, "exists", MagicMock(return_value=True))
|
||||
copy_sample_files(Path('/tmp/bar'))
|
||||
copy_sample_files(Path("/tmp/bar"))
|
||||
assert log_has_re(r"File `.*` exists already, not deploying sample file\.", caplog)
|
||||
caplog.clear()
|
||||
copy_sample_files(Path('/tmp/bar'), overwrite=True)
|
||||
copy_sample_files(Path("/tmp/bar"), overwrite=True)
|
||||
assert log_has_re(r"File `.*` exists already, overwriting\.", caplog)
|
||||
|
|
|
|||
|
|
@ -19,42 +19,42 @@ from freqtrade.loggers.set_log_levels import (
|
|||
|
||||
def test_set_loggers() -> None:
|
||||
# Reset Logging to Debug, otherwise this fails randomly as it's set globally
|
||||
logging.getLogger('requests').setLevel(logging.DEBUG)
|
||||
logging.getLogger("requests").setLevel(logging.DEBUG)
|
||||
logging.getLogger("urllib3").setLevel(logging.DEBUG)
|
||||
logging.getLogger('ccxt.base.exchange').setLevel(logging.DEBUG)
|
||||
logging.getLogger('telegram').setLevel(logging.DEBUG)
|
||||
logging.getLogger("ccxt.base.exchange").setLevel(logging.DEBUG)
|
||||
logging.getLogger("telegram").setLevel(logging.DEBUG)
|
||||
|
||||
previous_value1 = logging.getLogger('requests').level
|
||||
previous_value2 = logging.getLogger('ccxt.base.exchange').level
|
||||
previous_value3 = logging.getLogger('telegram').level
|
||||
previous_value1 = logging.getLogger("requests").level
|
||||
previous_value2 = logging.getLogger("ccxt.base.exchange").level
|
||||
previous_value3 = logging.getLogger("telegram").level
|
||||
|
||||
set_loggers()
|
||||
|
||||
value1 = logging.getLogger('requests').level
|
||||
value1 = logging.getLogger("requests").level
|
||||
assert previous_value1 is not value1
|
||||
assert value1 is logging.INFO
|
||||
|
||||
value2 = logging.getLogger('ccxt.base.exchange').level
|
||||
value2 = logging.getLogger("ccxt.base.exchange").level
|
||||
assert previous_value2 is not value2
|
||||
assert value2 is logging.INFO
|
||||
|
||||
value3 = logging.getLogger('telegram').level
|
||||
value3 = logging.getLogger("telegram").level
|
||||
assert previous_value3 is not value3
|
||||
assert value3 is logging.INFO
|
||||
|
||||
set_loggers(verbosity=2)
|
||||
|
||||
assert logging.getLogger('requests').level is logging.DEBUG
|
||||
assert logging.getLogger('ccxt.base.exchange').level is logging.INFO
|
||||
assert logging.getLogger('telegram').level is logging.INFO
|
||||
assert logging.getLogger('werkzeug').level is logging.INFO
|
||||
assert logging.getLogger("requests").level is logging.DEBUG
|
||||
assert logging.getLogger("ccxt.base.exchange").level is logging.INFO
|
||||
assert logging.getLogger("telegram").level is logging.INFO
|
||||
assert logging.getLogger("werkzeug").level is logging.INFO
|
||||
|
||||
set_loggers(verbosity=3, api_verbosity='error')
|
||||
set_loggers(verbosity=3, api_verbosity="error")
|
||||
|
||||
assert logging.getLogger('requests').level is logging.DEBUG
|
||||
assert logging.getLogger('ccxt.base.exchange').level is logging.DEBUG
|
||||
assert logging.getLogger('telegram').level is logging.INFO
|
||||
assert logging.getLogger('werkzeug').level is logging.ERROR
|
||||
assert logging.getLogger("requests").level is logging.DEBUG
|
||||
assert logging.getLogger("ccxt.base.exchange").level is logging.DEBUG
|
||||
assert logging.getLogger("telegram").level is logging.INFO
|
||||
assert logging.getLogger("werkzeug").level is logging.ERROR
|
||||
|
||||
|
||||
@pytest.mark.skipif(sys.platform == "win32", reason="does not run on windows")
|
||||
|
|
@ -63,9 +63,10 @@ def test_set_loggers_syslog():
|
|||
orig_handlers = logger.handlers
|
||||
logger.handlers = []
|
||||
|
||||
config = {'verbosity': 2,
|
||||
'logfile': 'syslog:/dev/log',
|
||||
}
|
||||
config = {
|
||||
"verbosity": 2,
|
||||
"logfile": "syslog:/dev/log",
|
||||
}
|
||||
|
||||
setup_logging_pre()
|
||||
setup_logging(config)
|
||||
|
|
@ -85,10 +86,11 @@ def test_set_loggers_Filehandler(tmp_path):
|
|||
logger = logging.getLogger()
|
||||
orig_handlers = logger.handlers
|
||||
logger.handlers = []
|
||||
logfile = tmp_path / 'ft_logfile.log'
|
||||
config = {'verbosity': 2,
|
||||
'logfile': str(logfile),
|
||||
}
|
||||
logfile = tmp_path / "ft_logfile.log"
|
||||
config = {
|
||||
"verbosity": 2,
|
||||
"logfile": str(logfile),
|
||||
}
|
||||
|
||||
setup_logging_pre()
|
||||
setup_logging(config)
|
||||
|
|
@ -111,9 +113,10 @@ def test_set_loggers_journald(mocker):
|
|||
orig_handlers = logger.handlers
|
||||
logger.handlers = []
|
||||
|
||||
config = {'verbosity': 2,
|
||||
'logfile': 'journald',
|
||||
}
|
||||
config = {
|
||||
"verbosity": 2,
|
||||
"logfile": "journald",
|
||||
}
|
||||
|
||||
setup_logging_pre()
|
||||
setup_logging(config)
|
||||
|
|
@ -129,11 +132,11 @@ def test_set_loggers_journald_importerror(import_fails):
|
|||
orig_handlers = logger.handlers
|
||||
logger.handlers = []
|
||||
|
||||
config = {'verbosity': 2,
|
||||
'logfile': 'journald',
|
||||
}
|
||||
with pytest.raises(OperationalException,
|
||||
match=r'You need the cysystemd python package.*'):
|
||||
config = {
|
||||
"verbosity": 2,
|
||||
"logfile": "journald",
|
||||
}
|
||||
with pytest.raises(OperationalException, match=r"You need the cysystemd python package.*"):
|
||||
setup_logging(config)
|
||||
logger.handlers = orig_handlers
|
||||
|
||||
|
|
@ -141,16 +144,16 @@ def test_set_loggers_journald_importerror(import_fails):
|
|||
def test_reduce_verbosity():
|
||||
setup_logging_pre()
|
||||
reduce_verbosity_for_bias_tester()
|
||||
prior_level = logging.getLogger('freqtrade').getEffectiveLevel()
|
||||
prior_level = logging.getLogger("freqtrade").getEffectiveLevel()
|
||||
|
||||
assert logging.getLogger('freqtrade.resolvers').getEffectiveLevel() == logging.WARNING
|
||||
assert logging.getLogger('freqtrade.strategy.hyper').getEffectiveLevel() == logging.WARNING
|
||||
assert logging.getLogger("freqtrade.resolvers").getEffectiveLevel() == logging.WARNING
|
||||
assert logging.getLogger("freqtrade.strategy.hyper").getEffectiveLevel() == logging.WARNING
|
||||
# base level wasn't changed
|
||||
assert logging.getLogger('freqtrade').getEffectiveLevel() == prior_level
|
||||
assert logging.getLogger("freqtrade").getEffectiveLevel() == prior_level
|
||||
|
||||
restore_verbosity_for_bias_tester()
|
||||
|
||||
assert logging.getLogger('freqtrade.resolvers').getEffectiveLevel() == prior_level
|
||||
assert logging.getLogger('freqtrade.strategy.hyper').getEffectiveLevel() == prior_level
|
||||
assert logging.getLogger('freqtrade').getEffectiveLevel() == prior_level
|
||||
assert logging.getLogger("freqtrade.resolvers").getEffectiveLevel() == prior_level
|
||||
assert logging.getLogger("freqtrade.strategy.hyper").getEffectiveLevel() == prior_level
|
||||
assert logging.getLogger("freqtrade").getEffectiveLevel() == prior_level
|
||||
# base level wasn't changed
|
||||
|
|
|
|||
|
|
@ -18,41 +18,34 @@ if sys.version_info < (3, 9):
|
|||
|
||||
def test_strategy_updater_start(user_dir, capsys) -> None:
|
||||
# Effective test without mocks.
|
||||
teststrats = Path(__file__).parent / 'strategy/strats'
|
||||
teststrats = Path(__file__).parent / "strategy/strats"
|
||||
tmpdirp = Path(user_dir) / "strategies"
|
||||
tmpdirp.mkdir(parents=True, exist_ok=True)
|
||||
shutil.copy(teststrats / 'strategy_test_v2.py', tmpdirp)
|
||||
old_code = (teststrats / 'strategy_test_v2.py').read_text()
|
||||
shutil.copy(teststrats / "strategy_test_v2.py", tmpdirp)
|
||||
old_code = (teststrats / "strategy_test_v2.py").read_text()
|
||||
|
||||
args = [
|
||||
"strategy-updater",
|
||||
"--userdir",
|
||||
str(user_dir),
|
||||
"--strategy-list",
|
||||
"StrategyTestV2"
|
||||
]
|
||||
args = ["strategy-updater", "--userdir", str(user_dir), "--strategy-list", "StrategyTestV2"]
|
||||
pargs = get_args(args)
|
||||
pargs['config'] = None
|
||||
pargs["config"] = None
|
||||
|
||||
start_strategy_update(pargs)
|
||||
|
||||
assert Path(user_dir / "strategies_orig_updater").exists()
|
||||
# Backup file exists
|
||||
assert Path(user_dir / "strategies_orig_updater" / 'strategy_test_v2.py').exists()
|
||||
assert Path(user_dir / "strategies_orig_updater" / "strategy_test_v2.py").exists()
|
||||
# updated file exists
|
||||
new_file = tmpdirp / 'strategy_test_v2.py'
|
||||
new_file = tmpdirp / "strategy_test_v2.py"
|
||||
assert new_file.exists()
|
||||
new_code = new_file.read_text()
|
||||
assert 'INTERFACE_VERSION = 3' in new_code
|
||||
assert 'INTERFACE_VERSION = 2' in old_code
|
||||
assert "INTERFACE_VERSION = 3" in new_code
|
||||
assert "INTERFACE_VERSION = 2" in old_code
|
||||
captured = capsys.readouterr()
|
||||
|
||||
assert 'Conversion of strategy_test_v2.py started.' in captured.out
|
||||
assert re.search(r'Conversion of strategy_test_v2\.py took .* seconds', captured.out)
|
||||
assert "Conversion of strategy_test_v2.py started." in captured.out
|
||||
assert re.search(r"Conversion of strategy_test_v2\.py took .* seconds", captured.out)
|
||||
|
||||
|
||||
def test_strategy_updater_methods(default_conf, caplog) -> None:
|
||||
|
||||
instance_strategy_updater = StrategyUpdater()
|
||||
modified_code1 = instance_strategy_updater.update_code("""
|
||||
class testClass(IStrategy):
|
||||
|
|
|
|||
|
|
@ -8,61 +8,61 @@ from freqtrade.exceptions import OperationalException
|
|||
|
||||
|
||||
def test_parse_timerange_incorrect():
|
||||
|
||||
timerange = TimeRange.parse_timerange('')
|
||||
timerange = TimeRange.parse_timerange("")
|
||||
assert timerange == TimeRange(None, None, 0, 0)
|
||||
timerange = TimeRange.parse_timerange('20100522-')
|
||||
assert TimeRange('date', None, 1274486400, 0) == timerange
|
||||
assert timerange.timerange_str == '20100522-'
|
||||
timerange = TimeRange.parse_timerange('-20100522')
|
||||
assert TimeRange(None, 'date', 0, 1274486400) == timerange
|
||||
assert timerange.timerange_str == '-20100522'
|
||||
timerange = TimeRange.parse_timerange('20100522-20150730')
|
||||
assert timerange == TimeRange('date', 'date', 1274486400, 1438214400)
|
||||
assert timerange.timerange_str == '20100522-20150730'
|
||||
assert timerange.start_fmt == '2010-05-22 00:00:00'
|
||||
assert timerange.stop_fmt == '2015-07-30 00:00:00'
|
||||
timerange = TimeRange.parse_timerange("20100522-")
|
||||
assert TimeRange("date", None, 1274486400, 0) == timerange
|
||||
assert timerange.timerange_str == "20100522-"
|
||||
timerange = TimeRange.parse_timerange("-20100522")
|
||||
assert TimeRange(None, "date", 0, 1274486400) == timerange
|
||||
assert timerange.timerange_str == "-20100522"
|
||||
timerange = TimeRange.parse_timerange("20100522-20150730")
|
||||
assert timerange == TimeRange("date", "date", 1274486400, 1438214400)
|
||||
assert timerange.timerange_str == "20100522-20150730"
|
||||
assert timerange.start_fmt == "2010-05-22 00:00:00"
|
||||
assert timerange.stop_fmt == "2015-07-30 00:00:00"
|
||||
|
||||
# Added test for unix timestamp - BTC genesis date
|
||||
assert TimeRange('date', None, 1231006505, 0) == TimeRange.parse_timerange('1231006505-')
|
||||
assert TimeRange(None, 'date', 0, 1233360000) == TimeRange.parse_timerange('-1233360000')
|
||||
timerange = TimeRange.parse_timerange('1231006505-1233360000')
|
||||
assert TimeRange('date', 'date', 1231006505, 1233360000) == timerange
|
||||
assert TimeRange("date", None, 1231006505, 0) == TimeRange.parse_timerange("1231006505-")
|
||||
assert TimeRange(None, "date", 0, 1233360000) == TimeRange.parse_timerange("-1233360000")
|
||||
timerange = TimeRange.parse_timerange("1231006505-1233360000")
|
||||
assert TimeRange("date", "date", 1231006505, 1233360000) == timerange
|
||||
assert isinstance(timerange.startdt, datetime)
|
||||
assert isinstance(timerange.stopdt, datetime)
|
||||
assert timerange.startdt == datetime.fromtimestamp(1231006505, tz=timezone.utc)
|
||||
assert timerange.stopdt == datetime.fromtimestamp(1233360000, tz=timezone.utc)
|
||||
assert timerange.timerange_str == '20090103-20090131'
|
||||
assert timerange.timerange_str == "20090103-20090131"
|
||||
|
||||
timerange = TimeRange.parse_timerange('1231006505000-1233360000000')
|
||||
assert TimeRange('date', 'date', 1231006505, 1233360000) == timerange
|
||||
timerange = TimeRange.parse_timerange("1231006505000-1233360000000")
|
||||
assert TimeRange("date", "date", 1231006505, 1233360000) == timerange
|
||||
|
||||
timerange = TimeRange.parse_timerange('1231006505000-')
|
||||
assert TimeRange('date', None, 1231006505, 0) == timerange
|
||||
timerange = TimeRange.parse_timerange("1231006505000-")
|
||||
assert TimeRange("date", None, 1231006505, 0) == timerange
|
||||
|
||||
timerange = TimeRange.parse_timerange('-1231006505000')
|
||||
assert TimeRange(None, 'date', 0, 1231006505) == timerange
|
||||
timerange = TimeRange.parse_timerange("-1231006505000")
|
||||
assert TimeRange(None, "date", 0, 1231006505) == timerange
|
||||
|
||||
with pytest.raises(OperationalException, match=r'Incorrect syntax.*'):
|
||||
TimeRange.parse_timerange('-')
|
||||
with pytest.raises(OperationalException, match=r"Incorrect syntax.*"):
|
||||
TimeRange.parse_timerange("-")
|
||||
|
||||
with pytest.raises(OperationalException,
|
||||
match=r'Start date is after stop date for timerange.*'):
|
||||
TimeRange.parse_timerange('20100523-20100522')
|
||||
with pytest.raises(
|
||||
OperationalException, match=r"Start date is after stop date for timerange.*"
|
||||
):
|
||||
TimeRange.parse_timerange("20100523-20100522")
|
||||
|
||||
|
||||
def test_subtract_start():
|
||||
x = TimeRange('date', 'date', 1274486400, 1438214400)
|
||||
x = TimeRange("date", "date", 1274486400, 1438214400)
|
||||
x.subtract_start(300)
|
||||
assert x.startts == 1274486400 - 300
|
||||
|
||||
# Do nothing if no startdate exists
|
||||
x = TimeRange(None, 'date', 0, 1438214400)
|
||||
x = TimeRange(None, "date", 0, 1438214400)
|
||||
x.subtract_start(300)
|
||||
assert not x.startts
|
||||
assert not x.startdt
|
||||
|
||||
x = TimeRange('date', None, 1274486400, 0)
|
||||
x = TimeRange("date", None, 1274486400, 0)
|
||||
x.subtract_start(300)
|
||||
assert x.startts == 1274486400 - 300
|
||||
|
||||
|
|
@ -70,17 +70,17 @@ def test_subtract_start():
|
|||
def test_adjust_start_if_necessary():
|
||||
min_date = datetime(2017, 11, 14, 21, 15, 00, tzinfo=timezone.utc)
|
||||
|
||||
x = TimeRange('date', 'date', 1510694100, 1510780500)
|
||||
x = TimeRange("date", "date", 1510694100, 1510780500)
|
||||
# Adjust by 20 candles - min_date == startts
|
||||
x.adjust_start_if_necessary(300, 20, min_date)
|
||||
assert x.startts == 1510694100 + (20 * 300)
|
||||
|
||||
x = TimeRange('date', 'date', 1510700100, 1510780500)
|
||||
x = TimeRange("date", "date", 1510700100, 1510780500)
|
||||
# Do nothing, startup is set and different min_date
|
||||
x.adjust_start_if_necessary(300, 20, min_date)
|
||||
assert x.startts == 1510694100 + (20 * 300)
|
||||
|
||||
x = TimeRange(None, 'date', 0, 1510780500)
|
||||
x = TimeRange(None, "date", 0, 1510780500)
|
||||
# Adjust by 20 candles = 20 * 5m
|
||||
x.adjust_start_if_necessary(300, 20, min_date)
|
||||
assert x.startts == 1510694100 + (20 * 300)
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user