use Test::Simple tests => 2;

use lib '../blib/lib','../blib/arch';

use Algorithm::DecisionTree;

# Test 1 (Digest Parameter File and Generate Training Data):

my $parameter_file = "t/param.txt";
my $output_training_datafile = "t/__trainingdata.csv";

my $data_gen = TrainingDataGeneratorSymbolic->new( 
                              output_training_datafile => $output_training_datafile,
                              parameter_file    => $parameter_file,
                              number_of_samples_for_training => 35,

               );
$data_gen->read_parameter_file_symbolic();
$data_gen->gen_symbolic_training_data();
open IN, $output_training_datafile;
my @data_records = <IN>;
ok( @data_records == 36,  'Training data created successfully' );

# Test 2 (Decision Tree Induction):

my $dt = Algorithm::DecisionTree->new(
                      training_datafile => $output_training_datafile,
                      csv_class_column_index => 1,
                      csv_columns_for_features => [2,3,4,5],
         );
$dt->get_training_data();
$dt->calculate_first_order_probabilities();
$dt->calculate_class_priors();
my $root_node = $dt->construct_decision_tree_classifier(); 
ok( defined($root_node),  'decision tree induction works' );

unlink "t/__trainingdata.csv";