Add an assert to expect_equal() to make sure people send in reasonable widths and...

[movit] / test_util.cpp

diff --git a/test_util.cpp b/test_util.cpp
index 111f8b2b21ce9773ea1a5ba34201286187b7f13c..b89b3ffb9ec64364768140b475ca53018b09a0a5 100644 (file)
--- a/test_util.cpp
+++ b/test_util.cpp
@@ -185,6 +185,9 @@ void expect_equal(const float *ref, const float *result, unsigned width, unsigne
 
 void expect_equal(const unsigned char *ref, const unsigned char *result, unsigned width, unsigned height, unsigned largest_difference_limit, float rms_limit)
 {
+       assert(width > 0);
+       assert(height > 0);
+
        float *ref_float = new float[width * height];
        float *result_float = new float[width * height];
 
@@ -196,4 +199,7 @@ void expect_equal(const unsigned char *ref, const unsigned char *result, unsigne
        }
 
        expect_equal(ref_float, result_float, width, height, largest_difference_limit, rms_limit);
+
+       delete[] ref_float;
+       delete[] result_float;
 }