conceptql 0.0.1

data/spec/conceptql/nodes/occurrence_spec.rb

@@ -0,0 +1,89 @@
+require 'spec_helper'
+require 'conceptql/nodes/occurrence'
+require_double('stream_for_occurrence')
+
+describe ConceptQL::Nodes::Occurrence do
+  it 'behaves itself' do
+    ConceptQL::Nodes::Occurrence.new.must_behave_like(:evaluator)
+  end
+
+  describe 'occurrence set to 1' do
+    subject do
+      ConceptQL::Nodes::Occurrence.new(1, StreamForOccurrenceDouble.new).query(Sequel.mock).sql
+    end
+
+    it 'should order by ascending start_date' do
+      subject.must_match 'ORDER BY start_date ASC'
+    end
+
+    it 'should partition by person_id' do
+      subject.must_match 'PARTITION BY person_id'
+    end
+
+    it 'should assign a row number' do
+      subject.must_match 'row_number()'
+    end
+
+    it 'should find the all rows with rn = 1' do
+      subject.must_match 'rn = 1'
+    end
+  end
+
+  describe 'occurrence set to 2' do
+    subject do
+      ConceptQL::Nodes::Occurrence.new(2, StreamForOccurrenceDouble.new).query(Sequel.mock).sql
+    end
+
+    it 'should order by ascending start_date' do
+      subject.must_match 'ORDER BY start_date ASC'
+    end
+
+    it 'should find the all rows with rn = 2' do
+      subject.must_match 'rn = 2'
+    end
+  end
+
+  describe 'occurrence set to -1' do
+    subject do
+      ConceptQL::Nodes::Occurrence.new(-1, StreamForOccurrenceDouble.new).query(Sequel.mock).sql
+    end
+
+    it 'should order by ascending start_date' do
+      subject.must_match 'ORDER BY start_date DESC'
+    end
+
+    it 'should find the all rows with rn = 1' do
+      subject.must_match 'rn = 1'
+    end
+  end
+
+  describe 'occurrence set to -2' do
+    subject do
+      ConceptQL::Nodes::Occurrence.new(-2, StreamForOccurrenceDouble.new).query(Sequel.mock).sql
+    end
+
+    it 'should order by ascending start_date' do
+      subject.must_match 'ORDER BY start_date DESC'
+    end
+
+    it 'should find the all rows with rn = 2' do
+      subject.must_match 'rn = 2'
+    end
+  end
+
+  describe 'occurrence respects types' do
+    subject do
+      dub = StreamForOccurrenceDouble.new
+      def dub.types
+        [:condition_occurrence]
+      end
+      ConceptQL::Nodes::Occurrence.new(-2, dub).query(Sequel.mock).sql
+    end
+
+    it 'should order by ascending start_date' do
+      subject.must_match ', condition_occurrence_id ASC'
+    end
+  end
+end
+
+

data/spec/conceptql/nodes/person_filter_spec.rb

@@ -0,0 +1,18 @@
+require 'spec_helper'
+require 'conceptql/nodes/person_filter'
+require_relative 'query_double'
+
+describe ConceptQL::Nodes::PersonFilter do
+  it 'behaves itself' do
+    ConceptQL::Nodes::PersonFilter.new.must_behave_like(:evaluator)
+  end
+
+  describe '#query' do
+    it 'uses right stream as argument to PERSON_FILTER against left stream' do
+      double1 = QueryDouble.new(1)
+      double2 = QueryDouble.new(2)
+      double1.must_behave_like(:evaluator)
+      ConceptQL::Nodes::PersonFilter.new(left: double1, right: double2).query(Sequel.mock).sql.must_equal "SELECT * FROM (SELECT * FROM table1) AS t1 WHERE (person_id IN (SELECT person_id FROM table2 GROUP BY person_id))"
+    end
+  end
+end

data/spec/conceptql/nodes/person_spec.rb

@@ -0,0 +1,12 @@
+require 'spec_helper'
+require 'conceptql/nodes/person'
+require_double('stream_for_casting')
+
+describe ConceptQL::Nodes::CastingNode do
+  it 'behaves itself' do
+    ConceptQL::Nodes::Person.new.must_behave_like(:casting_node)
+  end
+end
+
+
+

data/spec/conceptql/nodes/place_of_service_code_spec.rb

@@ -0,0 +1,26 @@
+require 'spec_helper'
+require 'conceptql/nodes/place_of_service_code'
+
+describe ConceptQL::Nodes::PlaceOfServiceCode do
+  it 'behaves itself' do
+    ConceptQL::Nodes::PlaceOfServiceCode.new.must_behave_like(:evaluator)
+  end
+
+  describe '#query' do
+    it 'works for 23' do
+      correct_query = "SELECT * FROM visit_occurrence_with_dates AS v INNER JOIN vocabulary.concept AS vc ON (vc.concept_id = v.place_of_service_concept_id) WHERE (vc.concept_code IN ('23'))"
+      ConceptQL::Nodes::PlaceOfServiceCode.new('23').query(Sequel.mock).sql.must_equal correct_query
+    end
+
+    it 'works for 23 as number' do
+      correct_query = "SELECT * FROM visit_occurrence_with_dates AS v INNER JOIN vocabulary.concept AS vc ON (vc.concept_id = v.place_of_service_concept_id) WHERE (vc.concept_code IN ('23'))"
+      ConceptQL::Nodes::PlaceOfServiceCode.new(23).query(Sequel.mock).sql.must_equal correct_query
+    end
+
+    it 'works for multiple values' do
+      correct_query = "SELECT * FROM visit_occurrence_with_dates AS v INNER JOIN vocabulary.concept AS vc ON (vc.concept_id = v.place_of_service_concept_id) WHERE (vc.concept_code IN ('23', '22'))"
+      ConceptQL::Nodes::PlaceOfServiceCode.new('23', '22').query(Sequel.mock).sql.must_equal correct_query
+    end
+  end
+end
+

data/spec/conceptql/nodes/procedure_occurrence_spec.rb

@@ -0,0 +1,12 @@
+require 'spec_helper'
+require 'conceptql/nodes/procedure_occurrence'
+require_double('stream_for_casting')
+
+describe ConceptQL::Nodes::CastingNode do
+  it 'behaves itself' do
+    ConceptQL::Nodes::ProcedureOccurrence.new.must_behave_like(:casting_node)
+  end
+end
+
+
+

data/spec/conceptql/nodes/query_double.rb

@@ -0,0 +1,19 @@
+require 'conceptql/nodes/node'
+class QueryDouble < ConceptQL::Nodes::Node
+  def initialize(num, type = :visit_occurrence)
+    @num = num
+    @type = type
+  end
+
+  def types
+    [@type]
+  end
+
+  def evaluate(db)
+    query(db)
+  end
+
+  def query(db)
+    db["table#{@num}".to_sym]
+  end
+end

data/spec/conceptql/nodes/race_spec.rb

@@ -0,0 +1,23 @@
+require 'spec_helper'
+require 'conceptql/nodes/race'
+
+describe ConceptQL::Nodes::Race do
+  it 'behaves itself' do
+    ConceptQL::Nodes::Race.new.must_behave_like(:evaluator)
+  end
+
+  describe '#query' do
+    it 'works for white' do
+      correct_query = "SELECT * FROM person_with_dates AS p INNER JOIN vocabulary.concept AS vc ON (vc.concept_id = p.race_concept_id) WHERE (lower(vc.concept_name) IN ('white'))"
+      ConceptQL::Nodes::Race.new('White').query(Sequel.mock).sql.must_equal correct_query
+      ConceptQL::Nodes::Race.new('white').query(Sequel.mock).sql.must_equal correct_query
+    end
+
+    it 'works for multiple values' do
+      correct_query = "SELECT * FROM person_with_dates AS p INNER JOIN vocabulary.concept AS vc ON (vc.concept_id = p.race_concept_id) WHERE (lower(vc.concept_name) IN ('white', 'other'))"
+      ConceptQL::Nodes::Race.new('White', 'Other').query(Sequel.mock).sql.must_equal correct_query
+      ConceptQL::Nodes::Race.new('white', 'other').query(Sequel.mock).sql.must_equal correct_query
+    end
+  end
+end
+

data/spec/conceptql/nodes/rxnorm_spec.rb

@@ -0,0 +1,31 @@
+require 'spec_helper'
+require 'conceptql/nodes/rxnorm'
+
+describe ConceptQL::Nodes::Rxnorm do
+  it 'behaves itself' do
+    ConceptQL::Nodes::Rxnorm.new.must_behave_like(:standard_vocabulary_node)
+  end
+
+  subject do
+    ConceptQL::Nodes::Rxnorm.new
+  end
+
+  describe '#table' do
+    it 'should be drug_exposure' do
+      subject.table.must_equal :drug_exposure
+    end
+  end
+
+  describe '#concept_column' do
+    it 'should be drug_concept_id' do
+      subject.concept_column.must_equal :drug_concept_id
+    end
+  end
+
+  describe '#vocabulary_id' do
+    it 'should be 8' do
+      subject.vocabulary_id.must_equal 8
+    end
+  end
+end
+

data/spec/conceptql/nodes/snomed_spec.rb

@@ -0,0 +1,31 @@
+require 'spec_helper'
+require 'conceptql/nodes/snomed'
+
+describe ConceptQL::Nodes::Snomed do
+  it 'behaves itself' do
+    ConceptQL::Nodes::Snomed.new.must_behave_like(:standard_vocabulary_node)
+  end
+
+  subject do
+    ConceptQL::Nodes::Snomed.new
+  end
+
+  describe '#table' do
+    it 'should be condition_occurrence' do
+      subject.table.must_equal :condition_occurrence
+    end
+  end
+
+  describe '#concept_column' do
+    it 'should be condition_concept_id' do
+      subject.concept_column.must_equal :condition_concept_id
+    end
+  end
+
+  describe '#vocabulary_id' do
+    it 'should be 1' do
+      subject.vocabulary_id.must_equal 1
+    end
+  end
+end
+

data/spec/conceptql/nodes/source_vocabulary_node_spec.rb

@@ -0,0 +1,37 @@
+require 'spec_helper'
+require 'conceptql/nodes/source_vocabulary_node'
+
+describe ConceptQL::Nodes::SourceVocabularyNode do
+  it 'behaves itself' do
+    ConceptQL::Nodes::SourceVocabularyNode.new.must_behave_like(:evaluator)
+  end
+
+  class SourceVocabularyDouble < ConceptQL::Nodes::SourceVocabularyNode
+    def table
+      :table
+    end
+
+    def source_column
+      :source_column
+    end
+
+    def concept_column
+      :concept_column
+    end
+
+    def vocabulary_id
+      1
+    end
+  end
+
+  describe '#query' do
+    it 'works for single values' do
+      SourceVocabularyDouble.new('value').query(Sequel.mock).sql.must_equal "SELECT * FROM table_with_dates AS tab INNER JOIN vocabulary.source_to_concept_map AS scm ON (scm.target_concept_id = tab.concept_column) WHERE ((scm.source_code IN ('value')) AND (scm.source_vocabulary_id = 1) AND (scm.source_code = tab.source_column))"
+    end
+
+    it 'works for multiple values' do
+      SourceVocabularyDouble.new('value1', 'value2').query(Sequel.mock).sql.must_equal "SELECT * FROM table_with_dates AS tab INNER JOIN vocabulary.source_to_concept_map AS scm ON (scm.target_concept_id = tab.concept_column) WHERE ((scm.source_code IN ('value1', 'value2')) AND (scm.source_vocabulary_id = 1) AND (scm.source_code = tab.source_column))"
+    end
+  end
+end
+

data/spec/conceptql/nodes/standard_vocabulary_node_spec.rb

@@ -0,0 +1,40 @@
+require 'spec_helper'
+require 'conceptql/nodes/standard_vocabulary_node'
+
+describe ConceptQL::Nodes::StandardVocabularyNode do
+  it 'behaves itself' do
+    ConceptQL::Nodes::StandardVocabularyNode.new.must_behave_like(:evaluator)
+  end
+
+  class StandardVocabularyDouble < ConceptQL::Nodes::StandardVocabularyNode
+    def table
+      :table
+    end
+
+    def concept_column
+      :concept_column
+    end
+
+    def vocabulary_id
+      1
+    end
+  end
+
+  describe StandardVocabularyDouble do
+    it 'behaves itself' do
+      StandardVocabularyDouble.new.must_behave_like(:standard_vocabulary_node)
+    end
+  end
+
+  describe '#query' do
+    it 'works for single values' do
+      StandardVocabularyDouble.new('value').query(Sequel.mock).sql.must_equal "SELECT * FROM table_with_dates AS tab INNER JOIN vocabulary.concept AS c ON (c.concept_id = tab.concept_column) WHERE ((c.concept_code IN ('value')) AND (c.vocabulary_id = 1))"
+    end
+
+    it 'works for multiple diagnoses' do
+      StandardVocabularyDouble.new('value1', 'value2').query(Sequel.mock).sql.must_equal "SELECT * FROM table_with_dates AS tab INNER JOIN vocabulary.concept AS c ON (c.concept_id = tab.concept_column) WHERE ((c.concept_code IN ('value1', 'value2')) AND (c.vocabulary_id = 1))"
+    end
+  end
+end
+
+

data/spec/conceptql/nodes/started_by_spec.rb

@@ -0,0 +1,25 @@
+require 'spec_helper'
+require 'conceptql/nodes/started_by'
+require_double('stream_for_temporal')
+
+describe ConceptQL::Nodes::StartedBy do
+  it 'behaves itself' do
+    ConceptQL::Nodes::StartedBy.new.must_behave_like(:temporal_node)
+  end
+
+  subject do
+    ConceptQL::Nodes::StartedBy.new(left: StreamForTemporalDouble.new, right: StreamForTemporalDouble.new)
+  end
+
+  it 'should use proper where clause' do
+    subject.query(Sequel.mock).sql.must_match 'l.start_date = r.start_date'
+    subject.query(Sequel.mock).sql.must_match 'l.end_date > r.end_date'
+  end
+
+  it 'should use proper where clause when inclusive' do
+    sub = ConceptQL::Nodes::StartedBy.new(left: StreamForTemporalDouble.new, right: StreamForTemporalDouble.new, inclusive: true)
+    sub.query(Sequel.mock).sql.must_match 'l.start_date = r.start_date'
+    sub.query(Sequel.mock).sql.must_match 'l.end_date >= r.end_date'
+  end
+end
+

data/spec/conceptql/nodes/temporal_node_spec.rb

@@ -0,0 +1,57 @@
+require 'spec_helper'
+require 'conceptql/nodes/temporal_node'
+require_double('stream_for_temporal')
+
+describe ConceptQL::Nodes::TemporalNode do
+  it 'behaves itself' do
+    ConceptQL::Nodes::TemporalNode.new.must_behave_like(:evaluator)
+  end
+
+  class TemporalDouble < ConceptQL::Nodes::TemporalNode
+    def where_clause
+      Proc.new do
+        l__end_date < r__start_date
+      end
+    end
+  end
+
+  describe TemporalDouble do
+    it 'behaves itself' do
+      TemporalDouble.new.must_behave_like(:temporal_node)
+    end
+  end
+
+  describe StreamForTemporalDouble do
+    it 'behaves itself' do
+      StreamForTemporalDouble.new.must_behave_like(:evaluator)
+    end
+  end
+
+  describe '#inclusive?' do
+    it 'defaults to false' do
+      refute(TemporalDouble.new.inclusive?)
+    end
+
+    it 'can be set to true' do
+      assert(TemporalDouble.new(inclusive: true).inclusive?)
+    end
+  end
+
+  describe '#query' do
+    it 'uses logic from where_clause' do
+      sql = TemporalDouble.new(left: StreamForTemporalDouble.new, right: StreamForTemporalDouble.new).query(Sequel.mock).sql
+      sql.must_match('l.end_date < r.start_date')
+    end
+
+    it 'pulls from the right tables' do
+      sql = TemporalDouble.new(left: StreamForTemporalDouble.new, right: StreamForTemporalDouble.new).query(Sequel.mock).sql
+      sql.must_match('FROM table')
+    end
+
+    it 'is ok with symbols' do
+      sql = TemporalDouble.new(left: StreamForTemporalDouble.new, right: StreamForTemporalDouble.new).query(Sequel.mock).sql
+      sql.must_match('FROM table')
+    end
+  end
+end
+

data/spec/conceptql/nodes/time_window_spec.rb

@@ -0,0 +1,66 @@
+require 'spec_helper'
+require 'conceptql/nodes/time_window'
+require 'conceptql/nodes/node'
+
+describe ConceptQL::Nodes::TimeWindow do
+  class Stream4TimeWindowDouble < ConceptQL::Nodes::Node
+    def types
+      [:visit_occurrence]
+    end
+
+    def query(db)
+      db
+    end
+  end
+
+  describe '#evaluate' do
+    it 'adjusts start by 1 day' do
+      sql = ConceptQL::Nodes::TimeWindow.new(Stream4TimeWindowDouble.new, { start: 'd', end: '' }).evaluate(Sequel.mock).sql
+      sql.must_match(%q{date((date(start_date) + interval '1 day')) AS start_date})
+      sql.must_match(%q{date(end_date) AS end_date})
+    end
+
+    it 'adjusts start by 1 day' do
+      sql = ConceptQL::Nodes::TimeWindow.new(Stream4TimeWindowDouble.new, { start: '', end: 'd' }).evaluate(Sequel.mock).sql
+      sql.must_match(%q{date(start_date) AS start_date})
+      sql.must_match(%q{date((date(end_date) + interval '1 day')) AS end_date})
+    end
+
+    it 'adjusts both values by 1 day' do
+      sql = ConceptQL::Nodes::TimeWindow.new(Stream4TimeWindowDouble.new, { start: 'd', end: 'd' }).evaluate(Sequel.mock).sql
+      sql.must_match(%q{date((date(start_date) + interval '1 day')) AS start_date, date((date(end_date) + interval '1 day')) AS end_date})
+    end
+
+    it 'makes multiple adjustments to both values' do
+      sql = ConceptQL::Nodes::TimeWindow.new(Stream4TimeWindowDouble.new, { start: 'dmy', end: '-d-m-y' }).evaluate(Sequel.mock).sql
+      sql.must_match(%q{date((date((date((date(start_date) + interval '1 day')) + interval '1 month')) + interval '1 year'))})
+      sql.must_match(%q{date((date((date((date(end_date) + interval '-1 day')) + interval '-1 month')) + interval '-1 year'))})
+    end
+
+    it 'can set start_date to be end_date' do
+      sql = ConceptQL::Nodes::TimeWindow.new(Stream4TimeWindowDouble.new, { start: 'end', end: '' }).evaluate(Sequel.mock).sql
+      sql.must_match(%q{end_date AS start_date})
+      sql.must_match(%q{date(end_date) AS end_date})
+    end
+
+    it 'can set end_date to be start_date' do
+      sql = ConceptQL::Nodes::TimeWindow.new(Stream4TimeWindowDouble.new, { start: '', end: 'start' }).evaluate(Sequel.mock).sql
+      sql.must_match(%q{start_date AS end_date})
+      sql.must_match(%q{date(start_date) AS start_date})
+    end
+
+    it 'will swap start and end dates, though this is a bad idea but you should probably know about this' do
+      sql = ConceptQL::Nodes::TimeWindow.new(Stream4TimeWindowDouble.new, { start: 'end', end: 'start' }).evaluate(Sequel.mock).sql
+      sql.must_match(%q{start_date AS end_date})
+      sql.must_match(%q{end_date AS start_date})
+    end
+
+    it 'handles nil arguments to both start and end' do
+      sql = ConceptQL::Nodes::TimeWindow.new(Stream4TimeWindowDouble.new, { start: nil, end: nil }).evaluate(Sequel.mock).sql
+      sql.must_match(%q{date(start_date) AS start_date})
+      sql.must_match(%q{date(end_date) AS end_date})
+    end
+  end
+end
+
+

data/spec/conceptql/nodes/union_spec.rb

@@ -0,0 +1,25 @@
+require 'spec_helper'
+require 'conceptql/nodes/union'
+require_relative 'query_double'
+
+describe ConceptQL::Nodes::Union do
+  it 'behaves itself' do
+    ConceptQL::Nodes::Union.new.must_behave_like(:evaluator)
+  end
+
+  describe '#query' do
+    it 'works for multiple criteria' do
+      double1 = QueryDouble.new(1)
+      double2 = QueryDouble.new(2)
+      double3 = QueryDouble.new(3)
+      double1.must_behave_like(:evaluator)
+      ConceptQL::Nodes::Union.new(double1, double2, double3).query(Sequel.mock).sql.must_equal "SELECT * FROM (SELECT * FROM (SELECT * FROM table1 UNION ALL SELECT * FROM table2) AS t1 UNION ALL SELECT * FROM table3) AS t1"
+    end
+
+    it 'works for single criteria' do
+      double1 = QueryDouble.new(1)
+      double1.must_behave_like(:evaluator)
+      ConceptQL::Nodes::Union.new(double1).query(Sequel.mock).sql.must_equal "SELECT * FROM table1"
+    end
+  end
+end

data/spec/conceptql/nodes/visit_occurrence_spec.rb

@@ -0,0 +1,12 @@
+require 'spec_helper'
+require 'conceptql/nodes/visit_occurrence'
+require_double('stream_for_casting')
+
+describe ConceptQL::Nodes::CastingNode do
+  it 'behaves itself' do
+    ConceptQL::Nodes::VisitOccurrence.new.must_behave_like(:casting_node)
+  end
+end
+
+
+

data/spec/conceptql/query_spec.rb

@@ -0,0 +1,20 @@
+require 'spec_helper'
+require 'conceptql'
+
+describe ConceptQL::Query do
+  describe '#query' do
+    it 'passes request on to tree' do
+      yaml = Psych.dump({ icd9: '799.22' })
+      mock_tree = Minitest::Mock.new
+      mock_node = Minitest::Mock.new
+
+      query = ConceptQL::Query.new(:mock_db, yaml, mock_tree)
+      mock_tree.expect :root, mock_node, [query]
+      mock_node.expect :evaluate, nil, [:mock_db]
+      query.query
+
+      mock_node.verify
+      mock_tree.verify
+    end
+  end
+end

data/spec/conceptql/tree_spec.rb

@@ -0,0 +1,54 @@
+require 'spec_helper'
+require 'conceptql/tree'
+
+describe ConceptQL::Tree do
+  describe '#root' do
+    before do
+      @mock_query_obj = Minitest::Mock.new
+      @mock_nodifier = Minitest::Mock.new
+      @tree = ConceptQL::Tree.new(nodifier: @mock_nodifier)
+    end
+
+    after do
+      @mock_query_obj.verify
+      @mock_nodifier.verify
+    end
+
+    it 'should walk single node criteria tree and convert to node' do
+      @mock_nodifier.expect :create, :success_indicator, [:icd9, '799.22']
+      @mock_query_obj.expect :statement, { icd9: '799.22' }
+
+      @tree.root(@mock_query_obj).must_equal :success_indicator
+    end
+
+    it 'should extend all nodes created with behavior passed in' do
+      mock_icd9_obj = Minitest::Mock.new
+      mock_icd9_obj.expect :extend, nil, [:mock_behavior]
+
+      @mock_nodifier.expect :create, mock_icd9_obj, [:icd9, '799.22']
+      @mock_query_obj.expect :statement, { icd9: '799.22' }
+
+      tree = ConceptQL::Tree.new(nodifier: @mock_nodifier, behavior: :mock_behavior)
+      tree.root(@mock_query_obj)
+    end
+
+    it 'should walk multi-criteria node' do
+      @mock_nodifier.expect :create, :mock_icd9, [:icd9, '799.22']
+      @mock_nodifier.expect :create, :success_indicator, [:nth, { occurrence: 1, expression: :mock_icd9 }]
+
+      @mock_query_obj.expect :statement, { nth: { occurrence: 1, expression: { icd9: '799.22' } } }
+
+      @tree.root(@mock_query_obj).must_equal :success_indicator
+    end
+
+    it 'should walk multi-node criteria tree and convert to nodes' do
+      @mock_nodifier.expect :create, :mock_icd9, [:icd9, '799.22']
+      @mock_nodifier.expect :create, :success_indicator, [:any, [:mock_icd9]]
+
+      @mock_query_obj.expect :statement, { any: [{ icd9: '799.22' }] }
+
+      @tree.root(@mock_query_obj).must_equal :success_indicator
+    end
+  end
+end
+

data/spec/doubles/stream_for_casting_double.rb

@@ -0,0 +1,9 @@
+class StreamForCastingDouble < ConceptQL::Nodes::Node
+  def query(db)
+    db.from(:table)
+  end
+
+  def types=(types)
+    @types = types
+  end
+end

data/spec/doubles/stream_for_occurrence_double.rb

@@ -0,0 +1,21 @@
+class StreamForOccurrenceDouble < ConceptQL::Nodes::Node
+  def query(db)
+    ds = db.from(:table)
+    # Occurrence needs window functions to work
+    meta_def(ds, :supports_window_functions?){true}
+    ds
+  end
+
+  # Stole this from:
+  # https://github.com/jeremyevans/sequel/blob/63397b787335d06de97dc89ddf49b7a3a93ffdc9/spec/core/expression_filters_spec.rb#L400
+  #
+  # By default, the Sequel.mock datasets don't allow window functions, but I need them
+  # enabled for testing
+  #
+  # I saw that Sequel tests had this little nugget in them to temporarily enable
+  # window functions and sure enough, it works
+  def meta_def(obj, name, &block)
+    (class << obj; self end).send(:define_method, name, &block)
+  end
+end
+

data/spec/doubles/stream_for_temporal_double.rb

@@ -0,0 +1,6 @@
+class StreamForTemporalDouble < ConceptQL::Nodes::Node
+  def query(db)
+    db.from(:table)
+  end
+end
+