@malloydata/malloy-tests 0.0.66-dev230807232724

package/src/models/medicare_model.ts

@@ -0,0 +1,246 @@
+/*
+ * Copyright 2023 Google LLC
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining
+ * a copy of this software and associated documentation files
+ * (the "Software"), to deal in the Software without restriction,
+ * including without limitation the rights to use, copy, modify, merge,
+ * publish, distribute, sublicense, and/or sell copies of the Software,
+ * and to permit persons to whom the Software is furnished to do so,
+ * subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be
+ * included in all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+ * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
+ * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
+ * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
+ * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ */
+
+import {StructDef} from '@malloydata/malloy';
+
+// will it build?
+
+/** Medicare Model */
+export const medicareModel: StructDef = {
+  as: 'medicare_test',
+  dialect: 'standardsql',
+  fields: [
+    // Fields in the flights table.
+    {name: 'id', numberType: 'integer', type: 'number'},
+    {name: 'drg_definition', type: 'string'},
+    {name: 'provider_id', numberType: 'integer', type: 'number'},
+    {name: 'provider_name', type: 'string'},
+    {name: 'provider_street', type: 'string'},
+    {name: 'provider_city', type: 'string'},
+    {name: 'provider_state', type: 'string'},
+    {name: 'provider_zipcode', numberType: 'integer', type: 'number'},
+    {name: 'hospital_referral_region_description', type: 'string'},
+    {
+      as: 'discharges',
+      name: 'total_discharges',
+      numberType: 'float',
+      type: 'number',
+    },
+    {
+      name: 'average_covered_charges',
+      numberType: 'float',
+      type: 'number',
+    },
+    {
+      name: 'average_total_payments',
+      numberType: 'float',
+      type: 'number',
+    },
+    {
+      name: 'average_medicare_payments',
+      numberType: 'float',
+      type: 'number',
+    },
+    {
+      type: 'number',
+      name: 'count_of_drugs',
+      expressionType: 'aggregate',
+      e: [{type: 'aggregate', function: 'count', e: []}],
+    },
+    {
+      type: 'number',
+      name: 'provider_count',
+      expressionType: 'aggregate',
+      e: ['COUNT(DISTINCT ', {type: 'field', path: 'provider_id'}, ')'],
+    },
+    {
+      type: 'number',
+      name: 'total_discharges',
+      expressionType: 'aggregate',
+      e: [
+        {
+          type: 'aggregate',
+          function: 'sum',
+          e: [{type: 'field', path: 'discharges'}],
+        },
+      ],
+    },
+
+    {
+      type: 'turtle',
+      name: 'discharges_by_state',
+      pipeline: [
+        {
+          fields: ['provider_state', 'total_discharges'],
+          orderBy: [{dir: 'desc', field: 2}],
+          type: 'reduce',
+        },
+      ],
+    },
+    {
+      type: 'turtle',
+      name: 'discharges_by_city',
+      pipeline: [
+        {
+          fields: ['provider_city', 'total_discharges'],
+          orderBy: [{dir: 'desc', field: 2}],
+          type: 'reduce',
+        },
+      ],
+    },
+    {
+      type: 'turtle',
+      name: 'bigturtle_state',
+      pipeline: [
+        {
+          fields: [
+            'provider_state',
+            'total_discharges',
+            'discharges_by_city',
+            'discharges_by_zip',
+          ],
+          orderBy: [{dir: 'desc', field: 1}],
+          type: 'reduce',
+        },
+      ],
+    },
+    {
+      type: 'turtle',
+      name: 'discharges_by_zip',
+      pipeline: [
+        {
+          fields: ['provider_zipcode', 'total_discharges'],
+          orderBy: [{dir: 'desc', field: 2}],
+          type: 'reduce',
+        },
+      ],
+    },
+    {
+      type: 'turtle',
+      name: 'turtle_city_zip',
+      pipeline: [
+        {
+          fields: ['provider_city', 'total_discharges', 'discharges_by_zip'],
+          orderBy: [{dir: 'desc', field: 1}],
+          type: 'reduce',
+        },
+      ],
+    },
+    {
+      type: 'turtle',
+      name: 'triple_turtle',
+      pipeline: [
+        {
+          fields: ['provider_state', 'total_discharges', 'turtle_city_zip'],
+          orderBy: [{dir: 'desc', field: 1}],
+          type: 'reduce',
+        },
+      ],
+    },
+    {
+      type: 'turtle',
+      name: 'rollup_by_location',
+      pipeline: [
+        {
+          fields: [
+            'provider_state',
+            'total_discharges',
+            {
+              type: 'turtle',
+              name: 'turtle_city_zip',
+              pipeline: [
+                {
+                  fields: [
+                    'provider_city',
+                    'total_discharges',
+                    {
+                      type: 'turtle',
+                      name: 'discharges_by_zip',
+                      pipeline: [
+                        {
+                          fields: ['provider_zipcode', 'total_discharges'],
+                          orderBy: [{dir: 'desc', field: 2}],
+                          type: 'reduce',
+                        },
+                      ],
+                    },
+                  ],
+                  orderBy: [{dir: 'desc', field: 1}],
+                  type: 'reduce',
+                },
+              ],
+            },
+          ],
+          orderBy: [{dir: 'desc', field: 1}],
+          type: 'reduce',
+        },
+      ],
+    },
+  ],
+  name: 'malloy-data.malloytest.bq_medicare_test',
+  primaryKey: 'id',
+  structRelationship: {type: 'basetable', connectionName: 'bigquery'},
+  structSource: {
+    type: 'table',
+    tablePath: 'malloy-data.malloytest.bq_medicare_test',
+  },
+  type: 'struct',
+};
+
+export const medicareStateFacts: StructDef = {
+  fields: [],
+  name: 'medicare_state_facts',
+  dialect: 'standardsql',
+  structRelationship: {type: 'basetable', connectionName: 'bigquery'},
+  structSource: {
+    query: {
+      structRef: 'medicare_test',
+      pipeline: [
+        {
+          fields: [
+            'provider_state',
+            {
+              type: 'number',
+              name: 'num_providers',
+              expressionType: 'aggregate',
+              e: ['COUNT(DISTINCT ', {type: 'field', path: 'provider_id'}, ')'],
+            },
+          ],
+          type: 'reduce',
+        },
+      ],
+    },
+    type: 'query',
+  },
+  type: 'struct',
+};
+
+// export const medicareStateFacts: StructDef = {
+//   type: 'struct',
+//   name: '(SELECT provider_state, COUNT(DISTINCT provider_id) as num_providers FROM malloytest.medicare_test GROUP BY 1)',
+//   as: 'medicare_state_facts',
+//   structRelationship: {type:'basetable'},
+//   structSource: {type:'table'},
+//   primaryKey: 'provider_state',
+//   fields: [{type:'string', name:'provider_state'}, {type:'number', name: 'num_providers', numberType: 'integer'}]
+// }