@malloydata/malloy-tests 0.0.94 → 0.0.95-dev231024220124

package/dist/databases/all/expr.spec.js

@@ -27,126 +27,106 @@ const runtimes_1 = require("../../runtimes");
 require("../../util/db-jest-matchers");
 const util_1 = require("../../util");
 const runtimes = new runtimes_1.RuntimeList((0, util_1.databasesFromEnvironmentOr)(runtimes_1.allDatabases));
-const expressionModelText = `
-source: aircraft_models is table('malloytest.aircraft_models') extend {
+function modelText(databaseName) {
+    return `
+source: aircraft_models is ${databaseName}.table('malloytest.aircraft_models') extend {
   primary_key: aircraft_model_code
   measure:
     aircraft_model_count is count(),
     total_seats is sum(seats),
-    boeing_seats is sum(seats) {? manufacturer ? 'BOEING'},
+    boeing_seats is sum(seats) { where: manufacturer ? 'BOEING'},
     percent_boeing is boeing_seats / total_seats * 100,
     percent_boeing_floor is floor(boeing_seats / total_seats * 100),
   dimension: seats_bucketed is floor(seats/20)*20.0
 }
 
-source: aircraft is table('malloytest.aircraft') extend {
+source: aircraft is ${databaseName}.table('malloytest.aircraft') extend {
   primary_key: tail_num
   join_one: aircraft_models with aircraft_model_code
   measure: aircraft_count is count()
-  query: by_manufacturer is {
+  view: by_manufacturer is {
     top: 5
     group_by: aircraft_models.manufacturer
     aggregate: aircraft_count
   }
 }
 `;
+}
 describe.each(runtimes.runtimeList)('%s', (databaseName, runtime) => {
-    const expressionModel = runtime.loadModel(expressionModelText);
+    const expressionModel = runtime.loadModel(modelText(databaseName));
     // basic calculations for sum, filtered sum, without a join.
     it('basic calculations', async () => {
-        const result = await expressionModel
-            .loadQuery(`
-        query: aircraft_models->{
-          aggregate:
-            total_seats,
-            total_seats2 is sum(seats),
-            boeing_seats,
-            boeing_seats2 is sum(seats) {? manufacturer ? 'BOEING'},
-            boeing_seats3 is total_seats {? manufacturer ? 'BOEING'},
-            percent_boeing,
-            percent_boeing2 is boeing_seats / total_seats * 100,
-            -- percent_boeing_floor,
-            -- percent_boeing_floor2 is FLOOR(boeing_seats / total_seats * 100)
-        }
-        `)
-            .run();
-        expect(result.data.path(0, 'total_seats').value).toBe(452415);
-        expect(result.data.path(0, 'total_seats2').value).toBe(452415);
-        expect(result.data.path(0, 'boeing_seats').value).toBe(252771);
-        expect(result.data.path(0, 'boeing_seats2').value).toBe(252771);
-        expect(result.data.path(0, 'boeing_seats3').value).toBe(252771);
-        expect(Math.floor(result.data.path(0, 'percent_boeing').number.value)).toBe(55);
-        expect(Math.floor(result.data.path(0, 'percent_boeing2').number.value)).toBe(55);
-        // expect(result.data.path(0, "percent_boeing_floor").value).toBe(55);
-        // expect(result.data.path(0, "percent_boeing_floor2").value).toBe(55);
+        await expect(`
+      run: aircraft_models->{
+        aggregate:
+          total_seats,
+          total_seats2 is sum(seats),
+          boeing_seats,
+          boeing_seats2 is sum(seats) { where: manufacturer ? 'BOEING'},
+          boeing_seats3 is total_seats { where: manufacturer ? 'BOEING'},
+          percent_boeing_floor,
+      }
+    `).malloyResultMatches(expressionModel, {
+            total_seats: 452415,
+            total_seats2: 452415,
+            boeing_seats: 252771,
+            boeing_seats2: 252771,
+            boeing_seats3: 252771,
+            percent_boeing_floor: 55,
+        });
     });
-    // Floor is broken (doesn't compile because the expression returned isn't an aggregate.)
+    // Floor was broken (wouldn't compile because the expression returned isn't an aggregate.)
     it('Floor() -or any function bustage with aggregates', async () => {
-        const result = await expressionModel
-            .loadQuery(`
-        query: aircraft_models->{
-          aggregate:
-            percent_boeing_floor
-            percent_boeing_floor2 is FLOOR(boeing_seats / total_seats * 100)
-        }
-      `)
-            .run();
-        expect(result.data.path(0, 'percent_boeing_floor').value).toBe(55);
-        expect(result.data.path(0, 'percent_boeing_floor2').value).toBe(55);
+        await expect(`
+      run: aircraft_models->{
+        aggregate:
+          percent_boeing_floor
+          percent_boeing_floor2 is floor(boeing_seats / total_seats * 100)
+      }
+    `).malloyResultMatches(expressionModel, {
+            percent_boeing_floor: 55,
+            percent_boeing_floor2: 55,
+        });
     });
     // Model based version of sums.
     it('model: expression fixups.', async () => {
-        const result = await expressionModel
-            .loadQuery(`
-            query: aircraft->{
-              aggregate:
-                aircraft_models.total_seats
-                aircraft_models.boeing_seats
-            }
-          `)
-            .run();
-        expect(result.data.path(0, 'total_seats').value).toBe(18294);
-        expect(result.data.path(0, 'boeing_seats').value).toBe(6244);
+        await expect(`
+      run: aircraft->{
+        aggregate:
+          aircraft_models.total_seats
+          aircraft_models.boeing_seats
+      }
+    `).malloyResultMatches(expressionModel, {
+            total_seats: 18294,
+            boeing_seats: 6244,
+        });
     });
     // turtle expressions
     it('model: turtle', async () => {
-        const result = await expressionModel
-            .loadQuery(`
-          query: aircraft->by_manufacturer
-          `)
-            .run();
-        expect(result.data.path(0, 'manufacturer').value).toBe('CESSNA');
+        await expect('run: aircraft->by_manufacturer').malloyResultMatches(expressionModel, { manufacturer: 'CESSNA' });
     });
     // filtered turtle expressions
     (0, util_1.testIf)(runtime.supportsNesting)('model: filtered turtle', async () => {
-        const result = await expressionModel
-            .loadQuery(`
-          query: aircraft->{
-            nest: b is by_manufacturer{? aircraft_models.manufacturer ?~'B%'}
-          }
-        `)
-            .run();
-        expect(result.data.path(0, 'b', 0, 'manufacturer').value).toBe('BEECH');
+        await expect(`
+      run: aircraft->{
+        nest: b is by_manufacturer refine { where: aircraft_models.manufacturer ?~'B%'}
+      }
+    `).malloyResultMatches(expressionModel, { 'b.manufacturer': 'BEECH' });
     });
     // having.
     it('model: simple having', async () => {
-        const result = await expressionModel
-            .loadQuery(`
-          query: aircraft->{
-            having: aircraft_count >90
-            group_by: state
-            aggregate: aircraft_count
-            order_by: 2
-          }
-          `)
-            .run();
-        expect(result.data.path(0, 'aircraft_count').value).toBe(91);
+        await expect(`
+      run: aircraft->{
+        having: aircraft_count >90
+        group_by: state
+        aggregate: aircraft_count
+        order_by: 2
+      }
+    `).malloyResultMatches(expressionModel, { aircraft_count: 91 });
     });
-    (0, util_1.testIf)(runtime.supportsNesting)('model: turtle having2', async () => {
-        const result = await expressionModel
-            .loadQuery(`
-      -- hacking a null test for now
-      query: aircraft->{
+    (0, util_1.testIf)(runtime.supportsNesting)('model: having in a nest', async () => {
+        await expect(`
+      run: aircraft->{
         top: 10
         order_by: 1
         where: region != NULL
@@ -159,14 +139,11 @@ describe.each(runtimes.runtimeList)('%s', (databaseName, runtime) => {
           aggregate: aircraft_count
         }
       }
-        `)
-            .run();
-        expect(result.data.path(0, 'by_state', 0, 'state').value).toBe('VA');
+    `).malloyResultMatches(expressionModel, { 'by_state.state': 'VA' });
     });
     (0, util_1.testIf)(runtime.supportsNesting)('model: turtle having on main', async () => {
-        const result = await expressionModel
-            .loadQuery(`
-      query: aircraft->{
+        await expect(`
+      run: aircraft->{
         order_by: 2 asc
         having: aircraft_count ? >500
         group_by: region
@@ -184,88 +161,86 @@ describe.each(runtimes.runtimeList)('%s', (databaseName, runtime) => {
           }
         }
       }
-        `)
-            .run();
-        expect(result.data.path(0, 'by_state', 0, 'by_city', 0, 'city').value).toBe('ALBUQUERQUE');
+    `).malloyResultMatches(expressionModel, {
+            'by_state.by_city.city': 'ALBUQUERQUE',
+        });
     });
     // bigquery doesn't like to partition by floats,
     (0, util_1.testIf)(runtime.supportsNesting)('model: having float group by partition', async () => {
-        await expect(runtime).queryMatches(`${expressionModelText}
-      query: aircraft_models->{
-        order_by: 1
-        where: seats_bucketed > 0
-        having: aircraft_model_count > 400
-        group_by: seats_bucketed
-        aggregate: aircraft_model_count
-        nest: foo is {
-          group_by: engines
+        await expect(`${modelText(databaseName)}
+        run: aircraft_models->{
+          order_by: 1
+          where: seats_bucketed > 0
+          having: aircraft_model_count > 400
+          group_by: seats_bucketed
           aggregate: aircraft_model_count
-        }
-      }`, { aircraft_model_count: 448 });
+          nest: foo is {
+            group_by: engines
+            aggregate: aircraft_model_count
+          }
+      }`).malloyResultMatches(runtime, { aircraft_model_count: 448 });
     });
     it('model: aggregate functions distinct min max', async () => {
-        const result = await expressionModel
-            .loadQuery(`
-        query: aircraft_models->{
-          aggregate:
-            distinct_seats is count(seats),
-            boeing_distinct_seats is count(seats) {?manufacturer ? 'BOEING'},
-            min_seats is min(seats),
-            cessna_min_seats is min(seats) {? manufacturer ? 'CESSNA'},
-            max_seats is max(seats),
-            cessna_max_seats is max(seats) {? manufacturer ? 'CESSNA'},
-            min_code is min(aircraft_model_code),
-            boeing_min_model is min(model) {? manufacturer ? 'BOEING'},
-            max_model is max(model),
-            boeing_max_model is max(model) {? manufacturer ? 'BOEING'},
-        }
-        `)
-            .run();
-        expect(result.data.path(0, 'distinct_seats').value).toBe(187);
-        expect(result.data.path(0, 'boeing_distinct_seats').value).toBe(85);
-        expect(result.data.path(0, 'min_seats').value).toBe(0);
-        expect(result.data.path(0, 'cessna_min_seats').value).toBe(1);
-        expect(result.data.path(0, 'max_seats').value).toBe(660);
-        expect(result.data.path(0, 'min_code').value).toBe('0030109');
-        expect(result.data.path(0, 'cessna_max_seats').value).toBe(14);
-        expect(result.data.path(0, 'boeing_min_model').value).toBe('100');
-        expect(result.data.path(0, 'max_model').value).toBe('ZWEIFEL PA18');
-        expect(result.data.path(0, 'boeing_max_model').value).toBe('YL-15');
+        await expect(`
+      run: aircraft_models->{
+        aggregate:
+          distinct_seats is count(seats),
+          boeing_distinct_seats is count(seats) { where:manufacturer ? 'BOEING'},
+          min_seats is min(seats),
+          cessna_min_seats is min(seats) { where: manufacturer ? 'CESSNA'},
+          max_seats is max(seats),
+          cessna_max_seats is max(seats) { where: manufacturer ? 'CESSNA'},
+          min_code is min(aircraft_model_code),
+          boeing_min_model is min(model) { where: manufacturer ? 'BOEING'},
+          max_model is max(model),
+          boeing_max_model is max(model) { where: manufacturer ? 'BOEING'},
+      }
+    `).malloyResultMatches(expressionModel, {
+            distinct_seats: 187,
+            boeing_distinct_seats: 85,
+            min_seats: 0,
+            cessna_min_seats: 1,
+            max_seats: 660,
+            min_code: '0030109',
+            cessna_max_seats: 14,
+            boeing_min_model: '100',
+            max_model: 'ZWEIFEL PA18',
+            boeing_max_model: 'YL-15',
+        });
     });
-    (databaseName !== 'bigquery' ? it.skip : it)('model: dates named', async () => {
-        const result = await expressionModel
-            .loadQuery(`
-        query: table('malloytest.alltypes')->{
-          group_by:
-            t_date,
-            t_date_month is t_date.month,
-            t_date_year is t_date.year,
-            t_timestamp,
-            t_timestamp_date is t_timestamp.day,
-            t_timestamp_hour is t_timestamp.hour,
-            t_timestamp_minute is t_timestamp.minute,
-            t_timestamp_second is t_timestamp.second,
-            t_timestamp_month is t_timestamp.month,
-            t_timestamp_year is t_timestamp.year,
-        }
-
-        `)
-            .run();
-        expect(result.data.path(0, 't_date').value).toEqual(new Date('2020-03-02'));
-        expect(result.data.path(0, 't_date_month').value).toEqual(new Date('2020-03-01'));
-        expect(result.data.path(0, 't_date_year').value).toEqual(new Date('2020-01-01'));
-        expect(result.data.path(0, 't_timestamp').value).toEqual(new Date('2020-03-02T12:35:56.000Z'));
-        expect(result.data.path(0, 't_timestamp_second').value).toEqual(new Date('2020-03-02T12:35:56.000Z'));
-        expect(result.data.path(0, 't_timestamp_minute').value).toEqual(new Date('2020-03-02T12:35:00.000Z'));
-        expect(result.data.path(0, 't_timestamp_hour').value).toEqual(new Date('2020-03-02T12:00:00.000Z'));
-        expect(result.data.path(0, 't_timestamp_date').value).toEqual(new Date('2020-03-02'));
-        expect(result.data.path(0, 't_timestamp_month').value).toEqual(new Date('2020-03-01'));
-        expect(result.data.path(0, 't_timestamp_year').value).toEqual(new Date('2020-01-01'));
+    // TODO not sure why this test needs to be skipped on postgres, feels like an oversight
+    (0, util_1.testIf)(databaseName !== 'postgres')('model: dates named', async () => {
+        await expect(`
+      run: ${databaseName}.table('malloytest.alltypes')->{
+        group_by:
+          t_date,
+          t_date_month is t_date.month,
+          t_date_year is t_date.year,
+          t_timestamp,
+          t_timestamp_date is t_timestamp.day,
+          t_timestamp_hour is t_timestamp.hour,
+          t_timestamp_minute is t_timestamp.minute,
+          t_timestamp_second is t_timestamp.second,
+          t_timestamp_month is t_timestamp.month,
+          t_timestamp_year is t_timestamp.year,
+      }
+    `).malloyResultMatches(runtime, {
+            t_date: new Date('2020-03-02'),
+            t_date_month: new Date('2020-03-01'),
+            t_date_year: new Date('2020-01-01'),
+            t_timestamp: new Date('2020-03-02T12:35:56.000Z'),
+            t_timestamp_second: new Date('2020-03-02T12:35:56.000Z'),
+            t_timestamp_minute: new Date('2020-03-02T12:35:00.000Z'),
+            t_timestamp_hour: new Date('2020-03-02T12:00:00.000Z'),
+            t_timestamp_date: new Date('2020-03-02'),
+            t_timestamp_month: new Date('2020-03-01'),
+            t_timestamp_year: new Date('2020-01-01'),
+        });
     });
     it('named query metadata undefined', async () => {
         const result = await expressionModel
             .loadQuery(`
-        query: aircraft->{
+        run: aircraft->{
           aggregate: aircraft_count is count()
         }
         `)
@@ -278,16 +253,14 @@ describe.each(runtimes.runtimeList)('%s', (databaseName, runtime) => {
     });
     it('named query metadata named', async () => {
         const result = await expressionModel
-            .loadQuery(`
-        query: aircraft->by_manufacturer
-        `)
+            .loadQuery('run: aircraft->by_manufacturer')
             .run();
         expect(result.resultExplore.name).toBe('by_manufacturer');
     });
     it('named query metadata named head of pipeline', async () => {
         const result = await expressionModel
             .loadQuery(`
-        query: aircraft->by_manufacturer->{ aggregate: c is count()}
+        run: aircraft->by_manufacturer->{ aggregate: c is count()}
         `)
             .run();
         // TODO Same as above -- this test should check the explore name
@@ -295,65 +268,51 @@ describe.each(runtimes.runtimeList)('%s', (databaseName, runtime) => {
         expect(result._queryResult.queryName).toBe(undefined);
     });
     it('filtered explores basic', async () => {
-        const result = await expressionModel
-            .loadQuery(`
-        source: b is aircraft{ where: aircraft_models.manufacturer ? ~'B%' }
-
-        query: b->{aggregate: m_count is count(aircraft_models.manufacturer) }
-        `)
-            .run();
-        expect(result.data.path(0, 'm_count').value).toBe(63);
+        await expect(`
+      run: aircraft extend { where: aircraft_models.manufacturer ? ~'B%' }
+        -> {aggregate: m_count is count(aircraft_models.manufacturer) }
+    `).malloyResultMatches(expressionModel, { m_count: 63 });
     });
     it('sql cast', async () => {
-        const result = await expressionModel
-            .loadQuery(`
-        query: aircraft -> {
-          group_by: a is "312"::"integer"
-        }
-        `)
-            .run();
-        expect(result.data.path(0, 'a').isNumber()).toBe(true);
-        expect(result.data.path(0, 'a').number.value).toBe(312);
-        if (runtime.connection.name !== 'postgres') {
-            const result = await expressionModel
-                .loadQuery(`
-          query: aircraft -> {
-            group_by: a is "312":::"integer"
-          }
-          `)
-                .run();
-            expect(result.data.path(0, 'a').isNumber()).toBe(true);
-            expect(result.data.path(0, 'a').number.value).toBe(312);
-        }
+        await expect(`
+      run: aircraft -> {
+        group_by: a is "312"::"integer"
+      }
+    `).malloyResultMatches(expressionModel, { a: 312 });
+    });
+    (0, util_1.testIf)(runtime.connection.name !== 'postgres')('sql safe cast', async () => {
+        await expect(`
+      run: ${databaseName}.sql('SELECT 1') -> { select:
+        bad_date is '123':::date
+        bad_number is 'abc':::number
+        good_number is "312":::"integer"
+      }
+    `).malloyResultMatches(expressionModel, {
+            bad_date: null,
+            bad_number: null,
+            good_number: 312,
+        });
     });
     it('many_field.sum() has correct locality', async () => {
-        const result = await expressionModel
-            .loadQuery(`
-
-        source: a is table('malloytest.aircraft')
-
-        source: am is table('malloytest.aircraft_models') extend {
-          join_many: a on a.aircraft_model_code = a.aircraft_model_code
-          dimension: a_year_built is a.year_built
-        }
-
-        // run: a -> {
-        //   aggregate: avg_year_built is avg(year_built)
-        // }
+        await expect(`
+      source: a is ${databaseName}.table('malloytest.aircraft')
+      source: am is ${databaseName}.table('malloytest.aircraft_models') extend {
+        join_many: a on a.aircraft_model_code = a.aircraft_model_code
+        dimension: a_year_built is a.year_built
+      }
 
-        run: am -> {
-          aggregate: avg_a_year_built1 is a_year_built.avg()
-          aggregate: avg_a_year_built2 is a.avg(a_year_built)
-        }
-        `)
-            .run();
-        expect(Math.floor(result.data.path(0, 'avg_a_year_built1').number.value)).toBe(1969);
-        expect(Math.floor(result.data.path(0, 'avg_a_year_built2').number.value)).toBe(1969);
+      run: am -> {
+        aggregate: avg_a_year_built1 is floor(a_year_built.avg())
+        aggregate: avg_a_year_built2 is floor(a.avg(a_year_built))
+      }
+    `).malloyResultMatches(runtime, {
+            avg_a_year_built1: 1969,
+            avg_a_year_built2: 1969,
+        });
     });
     (0, util_1.testIf)(runtime.supportsNesting)('query with aliasname used twice', async () => {
-        const result = await expressionModel
-            .loadQuery(`
-        query: aircraft->{
+        await expect(`
+        run: aircraft->{
           group_by: first is substr(city,1,1)
           aggregate: aircraft_count is count()
           nest: aircraft is {
@@ -370,145 +329,111 @@ describe.each(runtimes.runtimeList)('%s', (databaseName, runtime) => {
             aircraft_count
             order_by: 2 desc, 1
         }
-      `)
-            .run();
-        expect(result.data.path(0, 'first_three').value).toBe('SAB');
-    });
-    it.skip('join foreign_key reverse', async () => {
-        const result = await expressionModel
-            .loadQuery(`
-  source: a is table('malloytest.aircraft') {
-    primary_key: tail_num
-    measure: aircraft_count is count()
-  }
-  query: table('malloytest.aircraft_models') {
-    primary_key: aircraft_model_code
-    join_many: a on a.aircraft_model_code
-
-    some_measures is {
-      aggregate: am_count is count()
-      aggregate: a.aircraft_count
-    }
-  } -> some_measure
-    `)
-            .run();
-        expect(result.data.path(0, 'first_three').value).toBe('SAN');
+      `).malloyResultMatches(expressionModel, { first_three: 'SAB' });
     });
-    it('joined filtered explores', async () => {
-        const result = await expressionModel
-            .loadQuery(`
-    source: a_models is table('malloytest.aircraft_models'){
-      where: manufacturer ? ~'B%'
-      primary_key: aircraft_model_code
-      measure:model_count is count()
-    }
+    it('joined filtered sources', async () => {
+        await expect(`
+      source: a_models is ${databaseName}.table('malloytest.aircraft_models') extend {
+        where: manufacturer ? ~'B%'
+        primary_key: aircraft_model_code
+        measure:model_count is count()
+      }
 
-    source: aircraft2 is table('malloytest.aircraft'){
-      join_one: model is a_models with aircraft_model_code
-      measure: aircraft_count is count()
-    }
+      source: aircraft2 is ${databaseName}.table('malloytest.aircraft') extend {
+        join_one: model is a_models with aircraft_model_code
+        measure: aircraft_count is count()
+      }
 
-    query: aircraft2->{
-      aggregate:
-        model.model_count
-        aircraft_count
-    }
-        `)
-            .run();
-        expect(result.data.path(0, 'model_count').value).toBe(244);
-        expect(result.data.path(0, 'aircraft_count').value).toBe(3599);
+      run: aircraft2->{
+        aggregate:
+          model.model_count
+          aircraft_count
+      }
+    `).malloyResultMatches(expressionModel, {
+            model_count: 244,
+            aircraft_count: 3599,
+        });
     });
     it('joined filtered explores with dependancies', async () => {
-        const result = await expressionModel
-            .loadQuery(`
-    source: bo_models is
-      from(
-          table('malloytest.aircraft_models') {? manufacturer ? ~ 'BO%' }
-          -> { select: aircraft_model_code, manufacturer, seats }
-        ) {
+        await expect(`
+      source: bo_models is
+        ${databaseName}.table('malloytest.aircraft_models') extend { where: manufacturer ? ~ 'BO%' }
+        -> { select: aircraft_model_code, manufacturer, seats }
+        extend {
           primary_key: aircraft_model_code
           measure: bo_count is count()
         }
-
-    source: b_models is
-        from(
-          table('malloytest.aircraft_models') {? manufacturer ? ~ 'B%' }
-          -> { select: aircraft_model_code, manufacturer, seats }
-        ) {
+      source: b_models is
+        ${databaseName}.table('malloytest.aircraft_models') extend { where: manufacturer ? ~ 'B%' }
+        -> { select: aircraft_model_code, manufacturer, seats }
+        extend {
           where: bo_models.seats > 200
           primary_key: aircraft_model_code
           measure: b_count is count()
           join_one: bo_models with aircraft_model_code
         }
 
-    source: models is table('malloytest.aircraft_models') {
-      join_one: b_models with aircraft_model_code
-      measure: model_count is count()
-    }
+      source: models is ${databaseName}.table('malloytest.aircraft_models') extend {
+        join_one: b_models with aircraft_model_code
+        measure: model_count is count()
+      }
 
-    query: models -> {
-      aggregate: model_count
-      aggregate: b_models.b_count
-      -- aggregate: b_models.bo_models.bo_count
-    }
-        `)
-            .run();
-        expect(result.data.path(0, 'model_count').value).toBe(60461);
-        expect(result.data.path(0, 'b_count').value).toBe(355);
+      run: models -> {
+        aggregate: model_count
+        aggregate: b_models.b_count
+        -- aggregate: b_models.bo_models.bo_count
+      }
+    `).malloyResultMatches(runtime, { model_count: 60461, b_count: 355 });
     });
-    it('group by explore - simple group by', async () => {
-        const result = await expressionModel
-            .loadQuery(`
-        query: aircraft->{
-          group_by: aircraft_models
-          aggregate: aircraft_count
-        }
-    `)
-            .run();
-        expect(result.data.path(0, 'aircraft_count').value).toBe(58);
-        expect(result.data.path(0, 'aircraft_models_id').value).toBe('7102802');
+    it('group by join - simple group by', async () => {
+        await expect(`
+      run: aircraft->{
+        group_by: aircraft_models
+        aggregate: aircraft_count
+      }
+  `).malloyResultMatches(expressionModel, {
+            aircraft_models_id: '7102802',
+            aircraft_count: 58,
+        });
     });
     it('group by explore - pipeline', async () => {
-        const result = await expressionModel
-            .loadQuery(`
-        query: aircraft->{
-          group_by: aircraft_models
-          aggregate: aircraft_count
-        } -> {
-          group_by: aircraft_models.manufacturer
-          aggregate: aircraft_count is aircraft_count.sum()
-        }
-    `)
-            .run();
-        expect(result.data.path(0, 'aircraft_count').value).toBe(1048);
-        expect(result.data.path(0, 'manufacturer').value).toBe('CESSNA');
+        await expect(`
+      run: aircraft->{
+        group_by: aircraft_models
+        aggregate: aircraft_count
+      } -> {
+        group_by: aircraft_models.manufacturer
+        aggregate: aircraft_count is aircraft_count.sum()
+      }
+    `).malloyResultMatches(expressionModel, {
+            aircraft_count: 1048,
+            manufacturer: 'CESSNA',
+        });
     });
     it('group by explore - pipeline 2 levels', async () => {
-        const result = await expressionModel
-            .loadQuery(`
-      source: f is table('malloytest.flights'){
-        join_one: a is table('malloytest.aircraft') {
-          join_one: state_facts is table('malloytest.state_facts'){primary_key: state} with state
+        await expect(`
+      run: ${databaseName}.table('malloytest.flights') extend {
+        join_one: a is ${databaseName}.table('malloytest.aircraft') extend {
+          join_one: state_facts
+            is ${databaseName}.table('malloytest.state_facts') extend {primary_key: state}
+            with state
         } on tail_num = a.tail_num
-      }
-
-      query: f-> {
+      } -> {
         group_by: a.state_facts
         aggregate: flight_count is count()
       } -> {
         group_by: state_facts.popular_name
         aggregate: flight_count is flight_count.sum()
       }
-    `)
-            .run();
-        // console.log(result.data.toObject());
-        expect(result.data.path(0, 'flight_count').value).toBe(199726);
-        expect(result.data.path(0, 'popular_name').value).toBe('Isabella');
+    `).malloyResultMatches(runtime, {
+            flight_count: 199726,
+            popular_name: 'Isabella',
+        });
     });
 });
 describe.each(runtimes.runtimeList)('%s', (databaseName, runtime) => {
-    const sqlEq = (0, util_1.mkSqlEqWith)(runtime, {
-        malloy: `+ {
+    const sqlEq = (0, util_1.mkSqlEqWith)(runtime, databaseName, {
+        malloy: `extend {
       dimension: friName is 'friday'
       dimension: friDay is 5
       dimension: satName is 'saturday'
@@ -565,40 +490,38 @@ describe.each(runtimes.runtimeList)('%s', (databaseName, runtime) => {
             expect(await sqlEq(`'${back}${tick}'`, tick)).isSqlEq();
         });
         test('quote double quote', async () => {
-            await expect(runtime).queryMatches(`sql: x is {connection:"${databaseName}" select:"""SELECT 1 as x"""}
-        query: from_sql(x) -> {
+            await expect(`run: ${databaseName}.sql("SELECT 1") -> {
           select: double_quote is "${back}${dq}"
-        }
-      `, { double_quote: '"' });
+        }`).malloyResultMatches(runtime, { double_quote: '"' });
         });
         test('quote backslash', async () => {
             expect(await sqlEq(`'${back}${back}'`, back)).isSqlEq();
         });
     });
     test('nullish ?? operator', async () => {
-        await expect(runtime).queryMatches(`sql: nullTest is { connection: "${databaseName}" select: """
+        await expect(`run: ${databaseName}.sql("""
           SELECT '' as null_value, '' as string_value
           UNION ALL SELECT null, 'correct'
-      """ }
-      query: from_sql(nullTest) -> {
+      """) -> {
         where: null_value = null
         select:
           found_null is  null_value ?? 'correct',
           else_pass is string_value ?? 'incorrect'
           literal_null is null ?? 'correct'
-      }`, { found_null: 'correct', else_pass: 'correct', literal_null: 'correct' });
+      }`).malloyResultMatches(runtime, {
+            found_null: 'correct',
+            else_pass: 'correct',
+            literal_null: 'correct',
+        });
     });
     test('dimension expressions expanded with parens properly', async () => {
-        await expect(runtime).queryMatches(`
-      sql: tbl is { connection: "${databaseName}" select: """SELECT 1 as n"""}
-      query: from_sql(tbl) + {
+        await expect(`run: ${databaseName}.sql("SELECT 1") extend {
         dimension: fot is (false) or (true)
       } -> {
         select:
           no_paren is false and fot
           paren is    false and (fot)
-      }
-      `, { paren: false, no_paren: false });
+      }`).malloyResultMatches(runtime, { paren: false, no_paren: false });
     });
 });
 afterAll(async () => {