Implement test suite initialization and termination

[software/mutest.git] / sample / factorial_test.c

diff --git a/sample/factorial_test.c b/sample/factorial_test.c
index 10209e0b4a9478df633ef69328b7db7efad1eb61..0e0191a431578356680e5a2210e9a12c6e2fbd84 100644 (file)
--- a/sample/factorial_test.c
+++ b/sample/factorial_test.c
@@ -1,7 +1,12 @@
 
-#include "../mutest.h"
 #include "factorial.h"
 
+#ifdef MUTEST_PY
+#include "../py/mutest.h"
+#else
+#include "../mutest.h"
+#endif
+
 void mu_test_factorial_zero() {
        unsigned x = factorial(0);
        mu_check(x == 1);