benchmark: add assert map and set benchmarks

PR-URL: https://github.com/nodejs/node/pull/14258
Reviewed-By: Refael Ackermann <refack@gmail.com>

2 files changed, 255 insertions, 0 deletions

diff --git a/benchmark/assert/deepequal-map.js b/benchmark/assert/deepequal-map.js
new file mode 100644
index 0000000000..eae4a20605
--- /dev/null
+++ b/benchmark/assert/deepequal-map.js
@@ -0,0 +1,123 @@
+'use strict';
+
+/* eslint-disable no-restricted-properties */
+
+const common = require('../common.js');
+const assert = require('assert');
+
+const bench = common.createBenchmark(main, {
+  n: [5e2],
+  len: [5e2],
+  method: [
+    'deepEqual_primitiveOnly',
+    'deepStrictEqual_primitiveOnly',
+    'deepEqual_objectOnly',
+    'deepStrictEqual_objectOnly',
+    'deepEqual_mixed',
+    'deepStrictEqual_mixed',
+    'deepEqual_looseMatches',
+    'notDeepEqual_primitiveOnly',
+    'notDeepStrictEqual_primitiveOnly',
+    'notDeepEqual_objectOnly',
+    'notDeepStrictEqual_objectOnly',
+    'notDeepEqual_mixed',
+    'notDeepStrictEqual_mixed',
+    'notDeepEqual_looseMatches',
+  ]
+});
+
+function benchmark(method, n, values, values2) {
+  const actual = new Map(values);
+  // Prevent reference equal elements
+  const deepCopy = JSON.parse(JSON.stringify(values2 ? values2 : values));
+  const expected = new Map(deepCopy);
+  bench.start();
+  for (var i = 0; i < n; ++i) {
+    method(actual, expected);
+  }
+  bench.end(n);
+}
+
+function main(conf) {
+  const n = +conf.n;
+  const len = +conf.len;
+
+  const array = Array(len).fill(1);
+  var values, values2;
+
+  switch (conf.method) {
+    case 'deepEqual_primitiveOnly':
+      values = array.map((_, i) => [`str_${i}`, 123]);
+      benchmark(assert.deepEqual, n, values);
+      break;
+    case 'deepStrictEqual_primitiveOnly':
+      values = array.map((_, i) => [`str_${i}`, 123]);
+      benchmark(assert.deepStrictEqual, n, values);
+      break;
+    case 'deepEqual_objectOnly':
+      values = array.map((_, i) => [[`str_${i}`, 1], 123]);
+      benchmark(assert.deepEqual, n, values);
+      break;
+    case 'deepStrictEqual_objectOnly':
+      values = array.map((_, i) => [[`str_${i}`, 1], 123]);
+      benchmark(assert.deepStrictEqual, n, values);
+      break;
+    case 'deepEqual_mixed':
+      values = array.map((_, i) => [i % 2 ? [`str_${i}`, 1] : `str_${i}`, 123]);
+      benchmark(assert.deepEqual, n, values);
+      break;
+    case 'deepStrictEqual_mixed':
+      values = array.map((_, i) => [i % 2 ? [`str_${i}`, 1] : `str_${i}`, 123]);
+      benchmark(assert.deepStrictEqual, n, values);
+      break;
+    case 'deepEqual_looseMatches':
+      values = array.map((_, i) => [i, i]);
+      values2 = values.slice().map((v) => [String(v[0]), String(v[1])]);
+      benchmark(assert.deepEqual, n, values, values2);
+      break;
+    case 'notDeepEqual_primitiveOnly':
+      values = array.map((_, i) => [`str_${i}`, 123]);
+      values2 = values.slice(0);
+      values2[Math.floor(len / 2)] = ['w00t', 123];
+      benchmark(assert.notDeepEqual, n, values, values2);
+      break;
+    case 'notDeepStrictEqual_primitiveOnly':
+      values = array.map((_, i) => [`str_${i}`, 123]);
+      values2 = values.slice(0);
+      values2[Math.floor(len / 2)] = ['w00t', 123];
+      benchmark(assert.notDeepStrictEqual, n, values, values2);
+      break;
+    case 'notDeepEqual_objectOnly':
+      values = array.map((_, i) => [[`str_${i}`, 1], 123]);
+      values2 = values.slice(0);
+      values2[Math.floor(len / 2)] = [['w00t'], 123];
+      benchmark(assert.notDeepEqual, n, values, values2);
+      break;
+    case 'notDeepStrictEqual_objectOnly':
+      values = array.map((_, i) => [[`str_${i}`, 1], 123]);
+      values2 = values.slice(0);
+      values2[Math.floor(len / 2)] = [['w00t'], 123];
+      benchmark(assert.notDeepStrictEqual, n, values, values2);
+      break;
+    case 'notDeepEqual_mixed':
+      values = array.map((_, i) => [i % 2 ? [`str_${i}`, 1] : `str_${i}`, 123]);
+      values2 = values.slice(0);
+      values2[0] = ['w00t', 123];
+      benchmark(assert.notDeepEqual, n, values, values2);
+      break;
+    case 'notDeepStrictEqual_mixed':
+      values = array.map((_, i) => [i % 2 ? [`str_${i}`, 1] : `str_${i}`, 123]);
+      values2 = values.slice(0);
+      values2[0] = ['w00t', 123];
+      benchmark(assert.notDeepStrictEqual, n, values, values2);
+      break;
+    case 'notDeepEqual_looseMatches':
+      values = array.map((_, i) => [i, i]);
+      values2 = values.slice().map((v) => [String(v[0]), String(v[1])]);
+      values2[len - 1] = [String(len + 1), String(len + 1)];
+      benchmark(assert.notDeepEqual, n, values, values2);
+      break;
+    default:
+      throw new Error('Unsupported method');
+  }
+}
diff --git a/benchmark/assert/deepequal-set.js b/benchmark/assert/deepequal-set.js
new file mode 100644
index 0000000000..ecf44c063d
--- /dev/null
+++ b/benchmark/assert/deepequal-set.js
@@ -0,0 +1,132 @@
+'use strict';
+
+/* eslint-disable no-restricted-properties */
+
+const common = require('../common.js');
+const assert = require('assert');
+
+const bench = common.createBenchmark(main, {
+  n: [5e2],
+  len: [5e2],
+  method: [
+    'deepEqual_primitiveOnly',
+    'deepStrictEqual_primitiveOnly',
+    'deepEqual_objectOnly',
+    'deepStrictEqual_objectOnly',
+    'deepEqual_mixed',
+    'deepStrictEqual_mixed',
+    'deepEqual_looseMatches',
+    'notDeepEqual_primitiveOnly',
+    'notDeepStrictEqual_primitiveOnly',
+    'notDeepEqual_objectOnly',
+    'notDeepStrictEqual_objectOnly',
+    'notDeepEqual_mixed',
+    'notDeepStrictEqual_mixed',
+    'notDeepEqual_looseMatches',
+  ]
+});
+
+function benchmark(method, n, values, values2) {
+  const actual = new Set(values);
+  // Prevent reference equal elements
+  const deepCopy = JSON.parse(JSON.stringify(values2 ? values2 : values));
+  const expected = new Set(deepCopy);
+  bench.start();
+  for (var i = 0; i < n; ++i) {
+    method(actual, expected);
+  }
+  bench.end(n);
+}
+
+function main(conf) {
+  const n = +conf.n;
+  const len = +conf.len;
+
+  const array = Array(len).fill(1);
+
+  var values, values2;
+
+  switch (conf.method) {
+    case 'deepEqual_primitiveOnly':
+      values = array.map((_, i) => `str_${i}`);
+      benchmark(assert.deepEqual, n, values);
+      break;
+    case 'deepStrictEqual_primitiveOnly':
+      values = array.map((_, i) => `str_${i}`);
+      benchmark(assert.deepStrictEqual, n, values);
+      break;
+    case 'deepEqual_objectOnly':
+      values = array.map((_, i) => [`str_${i}`, null]);
+      benchmark(assert.deepEqual, n, values);
+      break;
+    case 'deepStrictEqual_objectOnly':
+      values = array.map((_, i) => [`str_${i}`, null]);
+      benchmark(assert.deepStrictEqual, n, values);
+      break;
+    case 'deepEqual_mixed':
+      values = array.map((_, i) => {
+        return i % 2 ? [`str_${i}`, null] : `str_${i}`;
+      });
+      benchmark(assert.deepEqual, n, values);
+      break;
+    case 'deepStrictEqual_mixed':
+      values = array.map((_, i) => {
+        return i % 2 ? [`str_${i}`, null] : `str_${i}`;
+      });
+      benchmark(assert.deepStrictEqual, n, values);
+      break;
+    case 'deepEqual_looseMatches':
+      values = array.map((_, i) => i);
+      values2 = values.slice().map((v) => String(v));
+      benchmark(assert.deepEqual, n, values, values2);
+      break;
+    case 'notDeepEqual_primitiveOnly':
+      values = array.map((_, i) => `str_${i}`);
+      values2 = values.slice(0);
+      values2[Math.floor(len / 2)] = 'w00t';
+      benchmark(assert.notDeepEqual, n, values, values2);
+      break;
+    case 'notDeepStrictEqual_primitiveOnly':
+      values = array.map((_, i) => `str_${i}`);
+      values2 = values.slice(0);
+      values2[Math.floor(len / 2)] = 'w00t';
+      benchmark(assert.notDeepStrictEqual, n, values, values2);
+      break;
+    case 'notDeepEqual_objectOnly':
+      values = array.map((_, i) => [`str_${i}`, null]);
+      values2 = values.slice(0);
+      values2[Math.floor(len / 2)] = ['w00t'];
+      benchmark(assert.notDeepEqual, n, values, values2);
+      break;
+    case 'notDeepStrictEqual_objectOnly':
+      values = array.map((_, i) => [`str_${i}`, null]);
+      values2 = values.slice(0);
+      values2[Math.floor(len / 2)] = ['w00t'];
+      benchmark(assert.notDeepStrictEqual, n, values, values2);
+      break;
+    case 'notDeepEqual_mixed':
+      values = array.map((_, i) => {
+        return i % 2 ? [`str_${i}`, null] : `str_${i}`;
+      });
+      values2 = values.slice();
+      values2[0] = 'w00t';
+      benchmark(assert.notDeepEqual, n, values, values2);
+      break;
+    case 'notDeepStrictEqual_mixed':
+      values = array.map((_, i) => {
+        return i % 2 ? [`str_${i}`, null] : `str_${i}`;
+      });
+      values2 = values.slice();
+      values2[0] = 'w00t';
+      benchmark(assert.notDeepStrictEqual, n, values, values2);
+      break;
+    case 'notDeepEqual_looseMatches':
+      values = array.map((_, i) => i);
+      values2 = values.slice().map((v) => String(v));
+      values2[len - 1] = String(len + 1);
+      benchmark(assert.notDeepEqual, n, values, values2);
+      break;
+    default:
+      throw new Error('Unsupported method');
+  }
+}