Skip to content

Commit

Permalink
going to update init
Browse files Browse the repository at this point in the history
  • Loading branch information
Ettel authored and Ettel committed Dec 20, 2023
1 parent 93ce4d6 commit 2aff4b1
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
3 changes: 2 additions & 1 deletion tests/helpers/_assertions.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,7 @@ def assert_that_tagged_tables_are_equal(table1: TaggedTable, table2: TaggedTable
assert table1.target == table2.target
assert table1 == table2


def assert_that_time_series_are_equal(table1: TimeSeries, table2: TimeSeries) -> None:
"""
Assert that two time series are equal.
Expand All @@ -55,4 +56,4 @@ def assert_that_time_series_are_equal(table1: TimeSeries, table2: TimeSeries) ->
assert table1.features == table2.features
assert table1.target == table2.target
assert table1.time == table2.time
assert table1 == table2
assert table1 == table2
Original file line number Diff line number Diff line change
Expand Up @@ -103,8 +103,8 @@ def test_should_raise_error(
error: type[Exception],
error_msg: str,
) -> None:
tagged_table = TaggedTable._from_table(table, target_name=target_name, feature_names=feature_names)
with pytest.raises(error, match=error_msg):
tagged_table = TaggedTable._from_table(table, target_name=target_name, feature_names=feature_names)
TimeSeries._from_tagged_table(tagged_table, time_name=time_name)


Expand Down

0 comments on commit 2aff4b1

Please sign in to comment.