Spec gbm data on correct class

lib/scoruby/models/gradient_boosted_model/data.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module Scoruby
   module Models
     module GradientBoostedModel

spec/scoruby/models/gradient_boosted_model/data_spec.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+describe Scoruby::Models::GradientBoostedModel::Data do
+  let(:xml) { Scoruby.xml_from_file_path(gbm_file) }
+  let(:gbm) { described_class.new(xml) }
+
+  context 'pmml 4.3' do
+    let(:gbm_file) { 'spec/fixtures/titanic_gbm.pmml' }
+
+    it 'loads correct number of trees' do
+      expect(gbm.decision_trees.count).to eq 100
+    end
+  end
+
+  context 'pmml 4.2' do
+    let(:gbm_file) { 'spec/fixtures/titanic_gbm_4_2.pmml' }
+
+    it 'loads correct number of trees' do
+      expect(gbm.decision_trees.count).to eq 100
+    end
+  end
+end

spec/scoruby/models/gradient_boosted_model/model_spec.rb

@@ -10,11 +10,7 @@
 
   context 'pmml 4.3' do
     let(:gbm_file) { 'spec/fixtures/titanic_gbm.pmml' }
-
-    it 'loads correct number of trees' do
-      expect(gbm.decision_trees.count).to eq 100
-    end
-
+
     it 'predicts approve' do
       expect(gbm.score(approve_features)).to eq 0.3652639329522468
     end
@@ -27,10 +23,6 @@
   context 'pmml 4.2' do
     let(:gbm_file) { 'spec/fixtures/titanic_gbm_4_2.pmml' }
 
-    it 'loads correct number of trees' do
-      expect(gbm.decision_trees.count).to eq 100
-    end
-
     it 'predicts approve' do
       expect(gbm.score(approve_features)).to eq 0.3652639329522468
     end