Args: ds (DataFlow): the DataFlow to test. size (int): number of datapoints to fetch. warmup (int): warmup iterations
(self, ds, size=5000, warmup=0)
| 29 | class TestDataSpeed(ProxyDataFlow): |
| 30 | """ Test the speed of a DataFlow """ |
| 31 | def __init__(self, ds, size=5000, warmup=0): |
| 32 | """ |
| 33 | Args: |
| 34 | ds (DataFlow): the DataFlow to test. |
| 35 | size (int): number of datapoints to fetch. |
| 36 | warmup (int): warmup iterations |
| 37 | """ |
| 38 | super(TestDataSpeed, self).__init__(ds) |
| 39 | self.test_size = int(size) |
| 40 | self.warmup = int(warmup) |
| 41 | self._reset_called = False |
| 42 | |
| 43 | def reset_state(self): |
| 44 | self._reset_called = True |