13
0
livetrax/libs/audiographer/tests/utils/identity_vertex_test.cc

101 lines
2.5 KiB
C++
Raw Normal View History

#include "tests/utils.h"
#include "audiographer/utils/identity_vertex.h"
using namespace AudioGrapher;
class IdentityVertexTest : public CppUnit::TestFixture
{
CPPUNIT_TEST_SUITE (IdentityVertexTest);
CPPUNIT_TEST (testProcess);
CPPUNIT_TEST (testRemoveOutput);
CPPUNIT_TEST (testClearOutputs);
CPPUNIT_TEST_SUITE_END ();
public:
void setUp()
{
samples = 128;
random_data = TestUtils::init_random_data(samples);
2015-10-05 10:17:49 -04:00
zero_data = new float[samples];
memset (zero_data, 0, samples * sizeof(float));
sink_a.reset (new VectorSink<float>());
sink_b.reset (new VectorSink<float>());
}
void tearDown()
{
delete [] random_data;
delete [] zero_data;
}
void testProcess()
{
vertex.reset (new IdentityVertex<float>());
vertex->add_output (sink_a);
vertex->add_output (sink_b);
2015-10-05 10:17:49 -04:00
samplecnt_t samples_output = 0;
2015-10-05 10:17:49 -04:00
ProcessContext<float> c (random_data, samples, 1);
vertex->process (c);
2015-10-05 10:17:49 -04:00
samples_output = sink_a->get_data().size();
CPPUNIT_ASSERT_EQUAL (samples, samples_output);
2015-10-05 10:17:49 -04:00
samples_output = sink_b->get_data().size();
CPPUNIT_ASSERT_EQUAL (samples, samples_output);
2015-10-05 10:17:49 -04:00
CPPUNIT_ASSERT (TestUtils::array_equals (random_data, sink_a->get_array(), samples));
CPPUNIT_ASSERT (TestUtils::array_equals (random_data, sink_b->get_array(), samples));
}
2015-10-05 10:17:49 -04:00
void testRemoveOutput()
{
vertex.reset (new IdentityVertex<float>());
vertex->add_output (sink_a);
vertex->add_output (sink_b);
2015-10-05 10:17:49 -04:00
ProcessContext<float> c (random_data, samples, 1);
vertex->process (c);
2015-10-05 10:17:49 -04:00
vertex->remove_output (sink_a);
ProcessContext<float> zc (zero_data, samples, 1);
vertex->process (zc);
2015-10-05 10:17:49 -04:00
CPPUNIT_ASSERT (TestUtils::array_equals (random_data, sink_a->get_array(), samples));
CPPUNIT_ASSERT (TestUtils::array_equals (zero_data, sink_b->get_array(), samples));
}
2015-10-05 10:17:49 -04:00
void testClearOutputs()
{
vertex.reset (new IdentityVertex<float>());
vertex->add_output (sink_a);
vertex->add_output (sink_b);
2015-10-05 10:17:49 -04:00
ProcessContext<float> c (random_data, samples, 1);
vertex->process (c);
2015-10-05 10:17:49 -04:00
vertex->clear_outputs ();
ProcessContext<float> zc (zero_data, samples, 1);
vertex->process (zc);
2015-10-05 10:17:49 -04:00
CPPUNIT_ASSERT (TestUtils::array_equals (random_data, sink_a->get_array(), samples));
CPPUNIT_ASSERT (TestUtils::array_equals (random_data, sink_b->get_array(), samples));
}
private:
boost::shared_ptr<IdentityVertex<float> > vertex;
boost::shared_ptr<VectorSink<float> > sink_a;
boost::shared_ptr<VectorSink<float> > sink_b;
float * random_data;
float * zero_data;
samplecnt_t samples;
};
CPPUNIT_TEST_SUITE_REGISTRATION (IdentityVertexTest);