Method setCurves

tests/backend/XYCurve/XYFunctionCurveTest.cpp:26–106

#include <QUndoStack>

void XYFunctionCurveTest::setCurves() {
	Project project;
	auto* ws = new Worksheet(QStringLiteral("Worksheet"));
	QVERIFY(ws != nullptr);
	project.addChild(ws);

	auto* p = new CartesianPlot(QStringLiteral("plot"));
	p->setType(CartesianPlot::Type::TwoAxes); // Otherwise no axis are created
	QVERIFY(p != nullptr);
	ws->addChild(p);

	p->addChild(new XYEquationCurve(QLatin1String("eq")));

	auto equationCurves = p->children(AspectType::XYEquationCurve);
	QCOMPARE(equationCurves.count(), 1);
	auto* equationCurve = static_cast<XYEquationCurve*>(equationCurves.at(0));
	XYEquationCurve::EquationData data;
	data.count = 100;
	data.expression1 = QStringLiteral("x");
	data.expression2 = QString();
	data.min = QStringLiteral("1");
	data.max = QStringLiteral("100");
	data.type = XYEquationCurve::EquationType::Cartesian;
	equationCurve->setEquationData(data);

	QCOMPARE(equationCurve->xColumn()->rowCount(), data.count);

	p->addChild(new XYFunctionCurve(QLatin1String("eq2")));
	auto functionCurves = p->children(AspectType::XYFunctionCurve);
	QCOMPARE(functionCurves.count(), 1);
	auto* functionCurve = static_cast<XYFunctionCurve*>(functionCurves.at(0));

	auto dock = XYFunctionCurveDock(nullptr);
	dock.setupGeneral();
	dock.setCurves({functionCurve});

	functionCurve->setFunction(QStringLiteral("2*x"), {QStringLiteral("x")}, {equationCurve});

	{
		const auto* xColumn = functionCurve->xColumn();
		const auto* yColumn = functionCurve->yColumn();
		QCOMPARE(xColumn->rowCount(), data.count);
		QCOMPARE(yColumn->rowCount(), data.count);
		for (int i = 0; i < xColumn->rowCount(); i++) {
			VALUES_EQUAL(xColumn->valueAt(i), i + 1.);
			VALUES_EQUAL(yColumn->valueAt(i), (i + 1.) * 2.);
		}
	}

	functionCurve->undoStack()->undo();

	QCOMPARE(functionCurve->xColumn()->rowCount(), 0);

	functionCurve->undoStack()->redo();

	{
		const auto* xColumn = functionCurve->xColumn();
		const auto* yColumn = functionCurve->yColumn();