Skip to content

Commit

Permalink
roc-streaminggh-752: Added more Tests for Moving Histogram Algorithm
Browse files Browse the repository at this point in the history
  • Loading branch information
spran180 authored and gavv committed Aug 1, 2024
1 parent 0bc19da commit 6d91f89
Showing 1 changed file with 140 additions and 0 deletions.
140 changes: 140 additions & 0 deletions src/tests/roc_core/test_mov_histogram.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -101,5 +101,145 @@ TEST(movhistogram, value_equal_to_value_range_max) {
LONGS_EQUAL(1, hist.get_bin_counter(expected_bin_index));
}

TEST(movhistogram, value_is_float) {
const float value_range_min = 0.0f;
const float value_range_max = 100.0f;
const size_t num_bins = 10;
const size_t win_length = 10;

MovHistogram<float> hist(arena, value_range_min, value_range_max, num_bins,
win_length);
CHECK(hist.is_valid());

for (size_t i = 0; i < win_length; i++) {
hist.add_value(i * num_bins * 1.0f);
}

for (size_t i = 0; i < num_bins; ++i) {
LONGS_EQUAL(1, hist.get_bin_counter(i));
}
}

TEST(movhistogram, min_max_negative) {
const int value_range_min = -150;
const int value_range_max = -50;
const size_t num_bins = 10;
const size_t win_length = 10;

MovHistogram<int> hist(arena, value_range_min, value_range_max, num_bins, win_length);
CHECK(hist.is_valid());

const int bin_width = (value_range_max - value_range_min) / (int)(num_bins);

for (size_t i = 0; i < win_length; i++) {
int value = value_range_min + (int)i * bin_width;
hist.add_value(value);
}

for (size_t i = 0; i < num_bins; ++i) {
LONGS_EQUAL(1, hist.get_bin_counter(i));
}
}

TEST(movhistogram, win_length_equal_one) {
const size_t value_range_min = 0;
const size_t value_range_max = 100;
const size_t num_bins = 10;
const size_t win_length = 1;

MovHistogram<size_t> hist(arena, value_range_min, value_range_max, num_bins,
win_length);
CHECK(hist.is_valid());

hist.add_value(0);
hist.add_value(10);
hist.add_value(20);

LONGS_EQUAL(0, hist.get_bin_counter(0));
LONGS_EQUAL(0, hist.get_bin_counter(1));
LONGS_EQUAL(1, hist.get_bin_counter(2));
}

TEST(movhistogram, multiple_values_in_bins) {
const size_t value_range_min = 0;
const size_t value_range_max = 100;
const size_t num_bins = 10;
const size_t win_length = 50;

MovHistogram<size_t> hist(arena, value_range_min, value_range_max, num_bins,
win_length);
CHECK(hist.is_valid());

const size_t values_per_bin = 5;
const size_t total_values = num_bins * values_per_bin;

for (size_t i = 0; i < total_values; ++i) {
size_t value = (i / values_per_bin) * (value_range_max / num_bins);
hist.add_value(value);
}

for (size_t i = 0; i < num_bins; ++i) {
LONGS_EQUAL(values_per_bin, hist.get_bin_counter(i));
}
}

TEST(movhistogram, rolling_window_bin_changes) {
const size_t value_range_min = 0;
const size_t value_range_max = 100;
const size_t num_bins = 10;
const size_t win_length = 5;

MovHistogram<size_t> hist(arena, value_range_min, value_range_max, num_bins,
win_length);
CHECK(hist.is_valid());

for (size_t i = 0; i < win_length; i++) {
hist.add_value(i * (value_range_max / num_bins));
}

for (size_t i = 0; i < num_bins; i++) {
LONGS_EQUAL(i < win_length ? 1 : 0, hist.get_bin_counter(i));
}

hist.add_value(win_length * (value_range_max / num_bins));

for (size_t i = 0; i < num_bins; i++) {
if (i < 1) {
LONGS_EQUAL(0, hist.get_bin_counter(i));
} else if (i <= win_length) {
LONGS_EQUAL(1, hist.get_bin_counter(i));
} else {
LONGS_EQUAL(0, hist.get_bin_counter(i));
}
}
}

TEST(movhistogram, clamping_values) {
const size_t value_range_min = 50;
const size_t value_range_max = 150;
const size_t num_bins = 10;
const size_t win_length = 10;

MovHistogram<size_t> hist(arena, value_range_min, value_range_max, num_bins,
win_length);
CHECK(hist.is_valid());

hist.add_value(static_cast<size_t>(20));
hist.add_value(static_cast<size_t>(5));
hist.add_value(static_cast<size_t>(10));

hist.add_value(static_cast<size_t>(60));
hist.add_value(static_cast<size_t>(80));

hist.add_value(static_cast<size_t>(160));
hist.add_value(static_cast<size_t>(170));
hist.add_value(static_cast<size_t>(180));

LONGS_EQUAL(3, hist.get_bin_counter(0));
LONGS_EQUAL(1, hist.get_bin_counter(1));
LONGS_EQUAL(1, hist.get_bin_counter(3));
LONGS_EQUAL(3, hist.get_bin_counter(9));
}

} // namespace core
} // namespace roc

0 comments on commit 6d91f89

Please sign in to comment.