test

vega · Dec 2, 2024 · fff4bef · fff4bef
1 parent ac3f4fd
commit fff4bef
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 5 deletions.
diff --git a/src/compile/data/timeunit.ts b/src/compile/data/timeunit.ts
@@ -10,7 +10,7 @@ import {
   normalizeTimeUnit
 } from '../../timeunit';
 import {TimeUnitTransform} from '../../transform';
-import {Dict, duplicate, entries, hash, isEmpty, replacePathInField, vals} from '../../util';
+import {Dict, duplicate, entries, escapePathAccess, hash, isEmpty, replacePathInField, vals} from '../../util';
 import {ModelWithField, isUnitModel} from '../model';
 import {DataFlowNode} from './dataflow';
 import {isRectBasedMark} from '../../mark';
@@ -218,7 +218,7 @@ function offsetExpr({timeUnit, field, reverse}: {timeUnit: TimeUnitParams; field
   const smallestUnit = getSmallestTimeUnitPart(unit);
   const {part, step} = getDateTimePartAndStep(smallestUnit, timeUnit.step);
   const offsetFn = utc ? 'utcOffset' : 'timeOffset';
-  const expr = `${offsetFn}('${part}', datum['${field}'], ${reverse ? -step : step})`;
+  const expr = `${offsetFn}('${part}', datum['${escapePathAccess(field)}'], ${reverse ? -step : step})`;
   return expr;
 }
 
@@ -228,8 +228,8 @@ function offsetedRectFormulas(
   timeUnit: TimeUnitParams
 ): VgFormulaTransform[] {
   if (rectBandPosition !== undefined && rectBandPosition !== 0.5) {
-    const startExpr = `datum['${startField}']`;
-    const endExpr = `datum['${endField}']`;
+    const startExpr = `datum['${escapePathAccess(startField)}']`;
+    const endExpr = `datum['${escapePathAccess(endField)}']`;
     return [
       {
         type: 'formula',

diff --git a/src/util.ts b/src/util.ts
@@ -293,7 +293,7 @@ export function flatAccessWithDatum(path: string, datum: 'datum' | 'parent' | 'd
   return `${datum}[${stringValue(splitAccessPath(path).join('.'))}]`;
 }
 
-function escapePathAccess(string: string) {
+export function escapePathAccess(string: string) {
   return string.replace(/(\[|\]|\.|'|")/g, '\\$1');
 }
 

diff --git a/test/compile/data/timeunit.test.ts b/test/compile/data/timeunit.test.ts
@@ -82,6 +82,36 @@ describe('compile/data/timeunit', () => {
       ]);
     });
 
+    it('should return a unit transform for bar with bandPosition=0 and escaped field name', () => {
+      const model = parseUnitModel({
+        data: {values: []},
+        mark: 'bar',
+        encoding: {
+          x: {field: "\\'a\\'", type: 'temporal', timeUnit: 'month', bandPosition: 0}
+        }
+      });
+
+      expect(assembleFromEncoding(model)).toEqual([
+        {
+          type: 'timeunit',
+          field: "\\'a\\'",
+          as: ["month_'a'", "month_'a'_end"],
+          units: ['month']
+        },
+        {
+          type: 'formula',
+          expr: "0.5 * timeOffset('month', datum['month_\\'a\\''], -1) + 0.5 * datum['month_\\'a\\'']",
+          as: `month_'a'_${OFFSETTED_RECT_START_SUFFIX}`
+        },
+
+        {
+          type: 'formula',
+          expr: "0.5 * datum['month_\\'a\\''] + 0.5 * datum['month_\\'a\\'_end']",
+          as: `month_'a'_${OFFSETTED_RECT_END_SUFFIX}`
+        }
+      ]);
+    });
+
     it('should return a timeunit transform with step for bar with bandPosition=0', () => {
       const model = parseUnitModel({
         data: {values: []},