mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-11-05 08:51:20 +00:00
869cfed125
* merge branch develop with main * bump desktop version * fix the merge * sync Cargo.lock to desktop 2.0.0 * fix cli conftest.py --------- Co-authored-by: deeleeramone <>
48 lines
1.1 KiB
Python
48 lines
1.1 KiB
Python
from datetime import date
|
|
|
|
import pytest
|
|
|
|
from openbb_famafrench.utils.missing_values import (
|
|
is_missing_value,
|
|
replace_missing_values,
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"value",
|
|
[None, float("nan"), -99.99, -999, -999.0, "-99.99", " -99.990", "-999", "-999.00"],
|
|
)
|
|
def test_is_missing_value_true(value):
|
|
assert is_missing_value(value) is True
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"value",
|
|
[
|
|
0.0,
|
|
-99.98,
|
|
-9.99,
|
|
12,
|
|
"1.5",
|
|
"-9.5",
|
|
"-9x",
|
|
"Lo 30",
|
|
"2020-01-31",
|
|
True,
|
|
date(2020, 1, 31),
|
|
],
|
|
)
|
|
def test_is_missing_value_false(value):
|
|
assert is_missing_value(value) is False
|
|
|
|
|
|
def test_replace_missing_values():
|
|
records = [
|
|
{"Date": "2020-01-31", "Mkt": "-99.99", "High": 1.5, "Firms": "12"},
|
|
{"Date": "2020-02-29", "Mkt": 2.0, "High": -999.0, "Firms": "-999"},
|
|
]
|
|
assert replace_missing_values(records, integer_fields=("Firms",)) == [
|
|
{"Date": "2020-01-31", "Mkt": None, "High": 1.5, "Firms": 12},
|
|
{"Date": "2020-02-29", "Mkt": 2.0, "High": None, "Firms": None},
|
|
]
|