mirror of
https://github.com/node-red/node-red-nodes.git
synced 2023-10-10 13:36:58 +02:00
b67452d569
* allow custom tokenizers allows use of tokens object to specifiy custom tokenizers updated mlsentiment to v2.0.0 (with temporary fix for comparative score applied) updated documentation * update unit tests
239 lines
9.7 KiB
JavaScript
239 lines
9.7 KiB
JavaScript
/**
|
|
* Copyright JS Foundation and other contributors, http://js.foundation
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
**/
|
|
|
|
var should = require("should");
|
|
var sentimentNode = require("../../../analysis/mlsentiment/mlsentiment.js");
|
|
var helper = require("node-red-node-test-helper");
|
|
|
|
describe('mlsentiment Node', function () {
|
|
|
|
before(function (done) {
|
|
helper.startServer(done);
|
|
});
|
|
|
|
after(function (done) {
|
|
helper.stopServer(done);
|
|
});
|
|
|
|
afterEach(function () {
|
|
helper.unload();
|
|
});
|
|
|
|
it('should be loaded', function (done) {
|
|
var flow = [{ id: "sentimentNode1", type: "mlsentiment", name: "sentimentNode" }];
|
|
helper.load(sentimentNode, flow, function () {
|
|
var sentimentNode1 = helper.getNode("sentimentNode1");
|
|
sentimentNode1.should.have.property('name', 'sentimentNode');
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should pass on msg if no payload', function (done) {
|
|
var flow = [{ id: "jn1", type: "mlsentiment", wires: [["jn2"]] },
|
|
{ id: "jn2", type: "helper" }];
|
|
helper.load(sentimentNode, flow, function () {
|
|
var jn1 = helper.getNode("jn1");
|
|
var jn2 = helper.getNode("jn2");
|
|
jn2.on("input", function (msg) {
|
|
msg.should.not.have.property('sentiment');
|
|
msg.topic.should.equal("pass on");
|
|
done();
|
|
});
|
|
var testString = 'good, great, best, brilliant';
|
|
jn1.receive({ topic: "pass on" });
|
|
});
|
|
});
|
|
|
|
it('should add a positive score for good words', function (done) {
|
|
var flow = [{ id: "jn1", type: "mlsentiment", wires: [["jn2"]] },
|
|
{ id: "jn2", type: "helper" }];
|
|
helper.load(sentimentNode, flow, function () {
|
|
var jn1 = helper.getNode("jn1");
|
|
var jn2 = helper.getNode("jn2");
|
|
jn2.on("input", function (msg) {
|
|
try {
|
|
msg.should.have.property('sentiment');
|
|
msg.sentiment.should.have.property('score');
|
|
msg.sentiment.score.should.be.a.Number();
|
|
msg.sentiment.score.should.be.above(10);
|
|
done();
|
|
} catch (err) {
|
|
done(err);
|
|
}
|
|
});
|
|
var testString = 'good, great, best, brilliant';
|
|
jn1.receive({ payload: testString });
|
|
});
|
|
});
|
|
|
|
it('should add a positive score for good words (in French)', function (done) {
|
|
var flow = [{ id: "jn1", type: "mlsentiment", wires: [["jn2"]], lang: "fr" },
|
|
{ id: "jn2", type: "helper" }];
|
|
helper.load(sentimentNode, flow, function () {
|
|
var jn1 = helper.getNode("jn1");
|
|
var jn2 = helper.getNode("jn2");
|
|
jn2.on("input", function (msg) {
|
|
try {
|
|
msg.should.have.property('sentiment');
|
|
msg.sentiment.should.have.property('score');
|
|
msg.sentiment.score.should.be.a.Number();
|
|
msg.sentiment.score.should.be.above(5);
|
|
done();
|
|
} catch (err) {
|
|
done(err);
|
|
}
|
|
});
|
|
var testString = 'bon, belle, don du ciel, brillant';
|
|
jn1.receive({ payload: testString });
|
|
});
|
|
});
|
|
|
|
it('should add a positive score for good words - alternative property', function (done) {
|
|
var flow = [{ id: "jn1", type: "mlsentiment", property: "foo", wires: [["jn2"]] },
|
|
{ id: "jn2", type: "helper" }];
|
|
helper.load(sentimentNode, flow, function () {
|
|
var jn1 = helper.getNode("jn1");
|
|
var jn2 = helper.getNode("jn2");
|
|
jn2.on("input", function (msg) {
|
|
try {
|
|
msg.should.have.property('sentiment');
|
|
msg.sentiment.should.have.property('score');
|
|
msg.sentiment.score.should.be.a.Number();
|
|
msg.sentiment.score.should.be.above(10);
|
|
done();
|
|
} catch (err) {
|
|
done(err);
|
|
}
|
|
});
|
|
var testString = 'good, great, best, brilliant';
|
|
jn1.receive({ foo: testString });
|
|
});
|
|
});
|
|
|
|
it('should add a negative score for bad words', function (done) {
|
|
var flow = [{ id: "jn1", type: "mlsentiment", wires: [["jn2"]] },
|
|
{ id: "jn2", type: "helper" }];
|
|
helper.load(sentimentNode, flow, function () {
|
|
var jn1 = helper.getNode("jn1");
|
|
var jn2 = helper.getNode("jn2");
|
|
jn2.on("input", function (msg) {
|
|
msg.should.have.property('sentiment');
|
|
msg.sentiment.should.have.property('score');
|
|
msg.sentiment.score.should.be.a.Number();
|
|
msg.sentiment.score.should.be.below(-10);
|
|
done();
|
|
});
|
|
var testString = 'bad, horrible, negative, awful';
|
|
jn1.receive({ payload: testString });
|
|
});
|
|
});
|
|
|
|
it('should add a negative score for bad words - alternative property', function (done) {
|
|
var flow = [{ id: "jn1", type: "mlsentiment", property: "foo", wires: [["jn2"]] },
|
|
{ id: "jn2", type: "helper" }];
|
|
helper.load(sentimentNode, flow, function () {
|
|
var jn1 = helper.getNode("jn1");
|
|
var jn2 = helper.getNode("jn2");
|
|
jn2.on("input", function (msg) {
|
|
msg.should.have.property('sentiment');
|
|
msg.sentiment.should.have.property('score');
|
|
msg.sentiment.score.should.be.a.Number();
|
|
msg.sentiment.score.should.be.below(-10);
|
|
done();
|
|
});
|
|
var testString = 'bad, horrible, negative, awful';
|
|
jn1.receive({ foo: testString });
|
|
});
|
|
});
|
|
|
|
it('should allow you to override word scoring', function (done) {
|
|
var flow = [{ id: "jn1", type: "mlsentiment", wires: [["jn2"]] },
|
|
{ id: "jn2", type: "helper" }];
|
|
helper.load(sentimentNode, flow, function () {
|
|
var jn1 = helper.getNode("jn1");
|
|
var jn2 = helper.getNode("jn2");
|
|
jn2.on("input", function (msg) {
|
|
msg.should.have.property('sentiment');
|
|
msg.sentiment.should.have.property('score');
|
|
msg.sentiment.score.should.be.a.Number();
|
|
msg.sentiment.score.should.equal(20);
|
|
done();
|
|
});
|
|
var testString = 'sick, wicked';
|
|
var wordOverrides = { 'sick': 10, 'wicked': 10 };
|
|
jn1.receive({ payload: testString, words: wordOverrides });
|
|
});
|
|
});
|
|
|
|
it('should allow you to override word scoring - alternative property', function (done) {
|
|
var flow = [{ id: "jn1", type: "mlsentiment", property: "foo", wires: [["jn2"]] },
|
|
{ id: "jn2", type: "helper" }];
|
|
helper.load(sentimentNode, flow, function () {
|
|
var jn1 = helper.getNode("jn1");
|
|
var jn2 = helper.getNode("jn2");
|
|
jn2.on("input", function (msg) {
|
|
msg.should.have.property('sentiment');
|
|
msg.sentiment.should.have.property('score');
|
|
msg.sentiment.score.should.be.a.Number();
|
|
msg.sentiment.score.should.equal(20);
|
|
done();
|
|
});
|
|
var testString = 'sick, wicked';
|
|
var wordOverrides = { 'sick': 10, 'wicked': 10 };
|
|
jn1.receive({ foo: testString, words: wordOverrides });
|
|
});
|
|
});
|
|
|
|
it('should allow you to use custom tokens', function (done) {
|
|
var flow = [{ id: "jn1", type: "mlsentiment", wires: [["jn2"]] },
|
|
{ id: "jn2", type: "helper" }];
|
|
helper.load(sentimentNode, flow, function () {
|
|
var jn1 = helper.getNode("jn1");
|
|
var jn2 = helper.getNode("jn2");
|
|
jn2.on("input", function (msg) {
|
|
msg.should.have.property('sentiment');
|
|
msg.sentiment.should.have.property('score');
|
|
msg.sentiment.score.should.be.a.Number();
|
|
msg.sentiment.score.should.equal(-3);
|
|
done();
|
|
});
|
|
var testString = '世界就是一个疯子的囚笼';
|
|
var tokenOverrides = ['世界', '就', '是', '一个', '疯子', '的', '囚笼'];
|
|
jn1.receive({ payload: testString, tokens: tokenOverrides });
|
|
});
|
|
});
|
|
|
|
it('should allow you to use custom tokens - alternative property', function (done) {
|
|
var flow = [{ id: "jn1", type: "mlsentiment", property: "foo", wires: [["jn2"]] },
|
|
{ id: "jn2", type: "helper" }];
|
|
helper.load(sentimentNode, flow, function () {
|
|
var jn1 = helper.getNode("jn1");
|
|
var jn2 = helper.getNode("jn2");
|
|
jn2.on("input", function (msg) {
|
|
msg.should.have.property('sentiment');
|
|
msg.sentiment.should.have.property('score');
|
|
msg.sentiment.score.should.be.a.Number();
|
|
msg.sentiment.score.should.equal(-3);
|
|
done();
|
|
});
|
|
var testString = '世界就是一个疯子的囚笼';
|
|
var tokenOverrides = ['世界', '就', '是', '一个', '疯子', '的', '囚笼'];
|
|
jn1.receive({ foo: testString, tokens: tokenOverrides });
|
|
});
|
|
});
|
|
|
|
});
|