13
0
livetrax/libs/audiographer/tests/general/deinterleaver_test.cc

130 lines
3.5 KiB
C++
Raw Normal View History

#include "tests/utils.h"
#include "audiographer/general/deinterleaver.h"
using namespace AudioGrapher;
class DeInterleaverTest : public CppUnit::TestFixture
{
CPPUNIT_TEST_SUITE (DeInterleaverTest);
CPPUNIT_TEST (testUninitialized);
CPPUNIT_TEST (testInvalidOutputIndex);
CPPUNIT_TEST (testInvalidInputSize);
CPPUNIT_TEST (testOutputSize);
CPPUNIT_TEST (testZeroInput);
CPPUNIT_TEST_SUITE_END ();
public:
void setUp()
{
channels = 3;
samples_per_channel = 128;
total_samples = channels * samples_per_channel;
random_data = TestUtils::init_random_data (total_samples, 1.0);
deinterleaver.reset (new DeInterleaver<float>());
sink_a.reset (new VectorSink<float>());
sink_b.reset (new VectorSink<float>());
sink_c.reset (new VectorSink<float>());
}
void tearDown()
{
delete [] random_data;
}
void testUninitialized()
{
deinterleaver.reset (new DeInterleaver<float>());
CPPUNIT_ASSERT_THROW (deinterleaver->output(0)->add_output (sink_a), Exception);
}
void testInvalidOutputIndex()
{
deinterleaver->init (3, samples_per_channel);
CPPUNIT_ASSERT_THROW (deinterleaver->output(3)->add_output (sink_a), Exception);
}
void testInvalidInputSize()
{
deinterleaver->init (channels, samples_per_channel);
2015-10-05 10:17:49 -04:00
ProcessContext<float> c (random_data, 2 * total_samples, channels);
2015-10-05 10:17:49 -04:00
// Too many, samples % channels == 0
CPPUNIT_ASSERT_THROW (deinterleaver->process (c.beginning (total_samples + channels)), Exception);
2015-10-05 10:17:49 -04:00
// Too many, samples % channels != 0
CPPUNIT_ASSERT_THROW (deinterleaver->process (c.beginning (total_samples + 1)), Exception);
2015-10-05 10:17:49 -04:00
// Too few, samples % channels != 0
CPPUNIT_ASSERT_THROW (deinterleaver->process (c.beginning (total_samples - 1)), Exception);
}
void assert_outputs (samplecnt_t expected_samples)
{
samplecnt_t generated_samples = 0;
2015-10-05 10:17:49 -04:00
generated_samples = sink_a->get_data().size();
CPPUNIT_ASSERT_EQUAL (expected_samples, generated_samples);
2015-10-05 10:17:49 -04:00
generated_samples = sink_b->get_data().size();
CPPUNIT_ASSERT_EQUAL (expected_samples, generated_samples);
2015-10-05 10:17:49 -04:00
generated_samples = sink_c->get_data().size();
CPPUNIT_ASSERT_EQUAL (expected_samples, generated_samples);
}
void testOutputSize()
{
deinterleaver->init (channels, samples_per_channel);
2015-10-05 10:17:49 -04:00
deinterleaver->output (0)->add_output (sink_a);
deinterleaver->output (1)->add_output (sink_b);
deinterleaver->output (2)->add_output (sink_c);
2015-10-05 10:17:49 -04:00
// Test maximum sample input
ProcessContext<float> c (random_data, total_samples, channels);
deinterleaver->process (c);
assert_outputs (samples_per_channel);
2015-10-05 10:17:49 -04:00
// Now with less samples
samplecnt_t const less_samples = samples_per_channel / 4;
deinterleaver->process (c.beginning (less_samples * channels));
assert_outputs (less_samples);
}
2015-10-05 10:17:49 -04:00
void testZeroInput()
{
deinterleaver->init (channels, samples_per_channel);
2015-10-05 10:17:49 -04:00
deinterleaver->output (0)->add_output (sink_a);
deinterleaver->output (1)->add_output (sink_b);
deinterleaver->output (2)->add_output (sink_c);
2015-10-05 10:17:49 -04:00
// Input zero samples
ProcessContext<float> c (random_data, total_samples, channels);
deinterleaver->process (c.beginning (0));
2015-10-05 10:17:49 -04:00
// ...and now test regular input
deinterleaver->process (c);
assert_outputs (samples_per_channel);
}
private:
boost::shared_ptr<DeInterleaver<float> > deinterleaver;
boost::shared_ptr<VectorSink<float> > sink_a;
boost::shared_ptr<VectorSink<float> > sink_b;
boost::shared_ptr<VectorSink<float> > sink_c;
float * random_data;
samplecnt_t samples_per_channel;
samplecnt_t total_samples;
unsigned int channels;
};
CPPUNIT_TEST_SUITE_REGISTRATION (DeInterleaverTest);