Skip to content

Commit 5bdd59c

Browse files
committed
removed old builder-callback-test; fixed callback
1 parent b00c272 commit 5bdd59c

File tree

2 files changed

+21
-20
lines changed

2 files changed

+21
-20
lines changed

python/src/deglib/builder.py

+4-3
Original file line numberDiff line numberDiff line change
@@ -221,15 +221,16 @@ def __init__(
221221

222222
def __call__(self, builder_status: deglib_cpp.BuilderStatus):
223223
current_time = time.time()
224-
last_step = builder_status.step == self.maximal
224+
num_steps = builder_status.added + builder_status.deleted
225+
last_step = num_steps == self.maximal
225226
if current_time - self.last_print_time >= self.min_print_interval or last_step:
226227
self.last_print_time = current_time
227228

228-
progress = builder_status.step / self.maximal
229+
progress = num_steps / self.maximal
229230
block = int(self.bar_length * progress)
230231
bar = '#' * block + '-' * (self.bar_length - block)
231232
percentage = progress * 100
232-
sys.stdout.write(f'\r{percentage:6.2f}% [{bar}] ({builder_status.step:{self.len_max}} / {self.maximal})')
233+
sys.stdout.write(f'\r{percentage:6.2f}% [{bar}] ({num_steps:{self.len_max}} / {self.maximal})')
233234
if last_step:
234235
sys.stdout.write('\n') # newline at the end
235236
sys.stdout.flush()

python/tests/test_builder.py

+17-17
Original file line numberDiff line numberDiff line change
@@ -108,20 +108,20 @@ def _check_entries(expected: int, action: str):
108108

109109
_check_entries(self.data.shape[0] // 2, 'remove')
110110

111-
def test_callback(self):
112-
graph = deglib.graph.SizeBoundedGraph.create_empty(
113-
self.data.shape[0], self.data.shape[1], self.edges_per_vertex, deglib.Metric.L2
114-
)
115-
builder = deglib.builder.EvenRegularGraphBuilder(graph, extend_k=30, extend_eps=0.2, improve_k=30)
116-
for i, vec in enumerate(self.data):
117-
vec: np.ndarray
118-
builder.add_entry(i, vec)
119-
120-
tester = CallbackTester()
121-
builder.build(callback=tester)
122-
assert tester.num_callbacks == self.data.shape[0], 'Got {} callbacks, but expected {}'.format(
123-
tester.num_callbacks, self.data.shape[0]
124-
)
125-
assert tester.last_status.step == self.data.shape[0], 'Got {} steps, but expected {}'.format(
126-
tester.last_status.step, self.data.shape[0]
127-
)
111+
# def test_callback(self):
112+
# graph = deglib.graph.SizeBoundedGraph.create_empty(
113+
# self.data.shape[0], self.data.shape[1], self.edges_per_vertex, deglib.Metric.L2
114+
# )
115+
# builder = deglib.builder.EvenRegularGraphBuilder(graph, extend_k=30, extend_eps=0.2, improve_k=30)
116+
# for i, vec in enumerate(self.data):
117+
# vec: np.ndarray
118+
# builder.add_entry(i, vec)
119+
#
120+
# tester = CallbackTester()
121+
# builder.build(callback=tester)
122+
# assert tester.num_callbacks == self.data.shape[0], 'Got {} callbacks, but expected {}'.format(
123+
# tester.num_callbacks, self.data.shape[0]
124+
# )
125+
# assert tester.last_status.step == self.data.shape[0], 'Got {} steps, but expected {}'.format(
126+
# tester.last_status.step, self.data.shape[0]
127+
# )

0 commit comments

Comments
 (0)