Skip to content

Commit

Permalink
Fix test coverage
Browse files Browse the repository at this point in the history
  • Loading branch information
aandres committed Jun 13, 2023
1 parent e3c4c2e commit 41938e9
Showing 1 changed file with 1 addition and 7 deletions.
8 changes: 1 addition & 7 deletions beavers/testing.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

import pandas as pd

from beavers.engine import UTC_MAX, Dag
from beavers.engine import Dag

T = TypeVar("T")

Expand All @@ -26,12 +26,6 @@ def run(self, now: Optional[pd.Timestamp] = None) -> "DagTestBench":
self.dag.stabilize(now)
return self

def run_for_next_timer(self) -> "DagTestBench":
next_timer = self.dag.get_next_timer()
assert next_timer != UTC_MAX
self.dag.stabilize(next_timer)
return self

def assert_sink_list(
self,
sink_name: str,
Expand Down

0 comments on commit 41938e9

Please sign in to comment.