1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
from maggma.core import Builder
class DummyBuilder(Builder):
def __init__(self, total=10):
self.get_called = 0
self.process_called = 0
self.update_called = 0
super().__init__(sources=[], targets=[])
self.total = total
def get_items(self):
for _i in range(self.total):
self.get_called += 1
yield self.get_called
def process_item(self, item):
self.process_called += 1
return item
def update_targets(self, items):
self.update_called += 1
__builder__ = DummyBuilder()
|
