Skip to content

Commit

Permalink
test: update test_strategies
Browse files Browse the repository at this point in the history
  • Loading branch information
hollandjg committed Nov 22, 2023
1 parent f5240f0 commit 6270044
Showing 1 changed file with 11 additions and 11 deletions.
22 changes: 11 additions & 11 deletions tests/test_strategies.py
Original file line number Diff line number Diff line change
Expand Up @@ -350,13 +350,18 @@ def dataframe_strategy(

df: pd.DataFrame = draw(
st_pd.data_frames(
columns=[st_pd.column(dtype=v.data_type) for v in variables],
columns=[st_pd.column(name=v.name, dtype=v.data_type) for v in variables],
)
)

return df


@given(dataframe_strategy())
def test_dataframe_strategy_creation(o):
assert o is not None


@st.composite
def model_strategy(draw, models=AVAILABLE_SKLEARN_MODELS_STRATEGY):
model = draw(models)
Expand All @@ -383,6 +388,11 @@ def model_strategy(draw, models=AVAILABLE_SKLEARN_MODELS_STRATEGY):
return result


@given(model_strategy())
def test_model_strategy_creation(o):
assert o


@st.composite
def standard_state_dataclass_strategy(draw):
variable_collection: VariableCollection = draw(variablecollection_strategy())
Expand All @@ -408,16 +418,6 @@ def standard_state_dataclass_strategy(draw):
return s


@given(dataframe_strategy())
def test_dataframe_strategy_creation(o):
assert o is not None


@given(model_strategy())
def test_model_strategy_creation(o):
assert o


@given(standard_state_dataclass_strategy())
def test_standard_state_dataclass_strategy_creation(o):
assert o
Expand Down

0 comments on commit 6270044

Please sign in to comment.