updated unit tests

fixed first trial handling in sticky and alternate too

Author: mdhumphries

Development/Test_interpolation_of_null_trials.m

@@ -3,8 +3,6 @@
 
 clearvars; close all;
 
-clearvars; close all;
-
 addpath ../Strategy_models/        % must add this path to access strategy models
 addpath ../Functions/
 

Development/UnitTest_strategy_models.m

@@ -0,0 +1,38 @@
+% script to unit test strategy models
+% latest version: 26/8/2022
+
+clearvars; close all;
+
+addpath ../Strategy_models/        % must add this path to access strategy models
+addpath ../Functions/
+
+% import test data CSV as Table
+TestData = readtable('UnitTestData.csv','TextType','string');
+
+% strategies to test
+strategies = ["go_left", "go_right", "go_cued","go_uncued",...
+                "win_stay_spatial","win_stay_cued","lose_shift_cued","lose_shift_spatial",...
+                   "alternate","sticky"];
+
+% storage
+TestResults = array2table(zeros(height(TestData),numel(strategies)));  
+TestResults.Properties.VariableNames = cellstr(strategies);
+
+% run all strategy models on Table data
+for iTrials = 1:height(TestData)
+    row_data = TestData(1:iTrials,:);
+    for index_strategy = 1:numel(strategies)
+        charStrategy = char(strategies(index_strategy)); % cast as Char for old MATLAB < 2018
+
+        % test current strategy model
+        trial_type = evaluate_strategy(strategies(index_strategy),row_data);
+        
+        % compare to target result
+        TestResults.(char(strategies(index_strategy)))(iTrials) = trial_type == TestData.(char(strategies(index_strategy))){iTrials};
+    end
+end
+
+% report any errors
+for index_strategy = 1:numel(strategies)
+    disp([string(TestResults.Properties.VariableNames{index_strategy}) ' passed? ' string(all(TestResults.(char(strategies(index_strategy)))))])
+end

Strategy_models/alternate.m

@@ -2,13 +2,15 @@
 
 % ALTERNATE checks if subject chose different spatial options on consecutive trials
 % TYPE = ALTERNATE(TRIAL_DATA) takes the Table of data TRIAL_DATA up to the current trial, and
-% returns the TYPE ('success','failure')
+% returns the TYPE ('success','failure'); returns 'null' for the first trial
 %
 % Mark Humphries 31/3/2022
 
 number_trials = size(trial_data,1);
 
-if number_trials > 1 && trial_data.Choice(end) ~= trial_data.Choice(end-1)
+if number_trials == 1
+    trial_type = "null";
+elseif trial_data.Choice(end) ~= trial_data.Choice(end-1)
     % chose different spatial option on consecutive trials, so is a success
     trial_type = "success";
 else

Strategy_models/sticky.m

@@ -2,13 +2,15 @@
 
 % STICKY checks if subject chose same spatial option on consecutive trials
 % TYPE = STICKY(TRIAL_DATA) takes the Table of data TRIAL_DATA up to the current trial, and
-% returns the TYPE ('success','failure')
+% returns the TYPE ('success','failure'); returns "null" on the first trial
 %
 % Mark Humphries 31/3/2022
 
 number_trials = size(trial_data,1);
 
-if number_trials > 1 && trial_data.Choice(end) == trial_data.Choice(end-1)
+if number_trials == 1 
+    trial_type = "null";
+elseif trial_data.Choice(end) == trial_data.Choice(end-1)
     % chose same spatial option on consecutive trials, so is a success
     trial_type = "success";
 else