Skip to content

Commit 806782d

Browse files
committed
askrene: add compiler flag ASKRENE_UNITTEST
Rusty: "We don't generally use NDEBUG in our code" Instead use a compile time flag ASKRENE_UNITTEST to make checks on unit tests that we don't normally need on release code. Changelog-none Signed-off-by: Lagrang3 <[email protected]>
1 parent d24e60a commit 806782d

File tree

8 files changed

+10
-4
lines changed

8 files changed

+10
-4
lines changed

plugins/askrene/algorithm.c

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -402,7 +402,7 @@ static struct node dijkstra_nearest_sink(const tal_t *ctx,
402402
prev[i].idx = INVALID_INDEX;
403403

404404
/* Only in debug mode we keep track of visited nodes. */
405-
#ifndef NDEBUG
405+
#ifdef ASKRENE_UNITTEST
406406
bitmap *visited =
407407
tal_arrz(this_ctx, bitmap, BITMAP_NWORDS(max_num_nodes));
408408
assert(visited);
@@ -421,7 +421,7 @@ static struct node dijkstra_nearest_sink(const tal_t *ctx,
421421
priorityqueue_pop(q);
422422

423423
/* Only in debug mode we keep track of visited nodes. */
424-
#ifndef NDEBUG
424+
#ifdef ASKRENE_UNITTEST
425425
assert(!bitmap_test_bit(visited, cur.idx));
426426
bitmap_set_bit(visited, cur.idx);
427427
#endif
@@ -598,7 +598,7 @@ bool mcf_refinement(const tal_t *ctx,
598598
}
599599
}
600600

601-
#ifndef NDEBUG
601+
#ifdef ASKRENE_UNITTEST
602602
/* verify that we have satisfied all constraints */
603603
for (u32 i = 0; i < max_num_nodes; i++) {
604604
assert(excess[i] == 0);

plugins/askrene/test/run-bfs.c

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
#include <plugins/askrene/graph.h>
77
#include <stdio.h>
88

9+
#define ASKRENE_UNITTEST
910
#include "../algorithm.c"
1011

1112
#define MAX_NODES 256

plugins/askrene/test/run-dijkstra.c

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
#include <plugins/askrene/graph.h>
77
#include <stdio.h>
88

9+
#define ASKRENE_UNITTEST
910
#include "../algorithm.c"
1011

1112
// 1->2 7

plugins/askrene/test/run-flow.c

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
#include <plugins/askrene/graph.h>
77
#include <stdio.h>
88

9+
#define ASKRENE_UNITTEST
910
#include "../algorithm.c"
1011

1112
#define MAX_NODES 256

plugins/askrene/test/run-graph.c

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
#include <inttypes.h>
77
#include <stdio.h>
88

9+
#define ASKRENE_UNITTEST
910
#include "../graph.c"
1011

1112
#define MAX_NODES 10

plugins/askrene/test/run-mcf-large.c

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
#include <plugins/askrene/graph.h>
77
#include <stdio.h>
88

9+
#define ASKRENE_UNITTEST
910
#include "../algorithm.c"
1011

1112
#ifdef HAVE_ZLIB

plugins/askrene/test/run-mcf.c

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
#include <plugins/askrene/graph.h>
77
#include <stdio.h>
88

9+
#define ASKRENE_UNITTEST
910
#include "../algorithm.c"
1011

1112
#define CHECK(arg) if(!(arg)){fprintf(stderr, "failed CHECK at line %d: %s\n", __LINE__, #arg); abort();}

plugins/askrene/test/run-pqueue.c

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
#include <inttypes.h>
66
#include <stdio.h>
77

8-
8+
#define ASKRENE_UNITTEST
99
#include "../priorityqueue.c"
1010

1111
#define CHECK(arg) if(!(arg)){fprintf(stderr, "failed CHECK at line %d: %s\n", __LINE__, #arg); abort();}

0 commit comments

Comments
 (0)