diff --git a/MarketDataLib/Generator/CMMomentum/CMMomentumGenerator.cs b/MarketDataLib/Generator/CMMomentum/CMMomentumGenerator.cs index 4ca95e1..48c3cfc 100644 --- a/MarketDataLib/Generator/CMMomentum/CMMomentumGenerator.cs +++ b/MarketDataLib/Generator/CMMomentum/CMMomentumGenerator.cs @@ -118,7 +118,8 @@ namespace MarketData.Generator.CMMomentum } return true; } - private static void PredictCandidate(CMCandidate cmCandidate,CMParams cmParams) + // This method is made public in order that it can be tested + public static bool PredictCandidate(CMCandidate cmCandidate,CMParams cmParams) { try { @@ -138,12 +139,12 @@ namespace MarketData.Generator.CMMomentum cmCandidate.Score*=(1.00+cmParams.UseCNNRewardPercentDecimal); // increase the score by the percentage indicated in the params settings cmCandidate.CNNPrediction=true; } - return; + return true; } catch(Exception exception) { MDTrace.WriteLine(LogLevel.DEBUG,String.Format("Error encountered calling convolutional model at {0}. Exception was {1}",cmParams.UseCNNHost,exception.ToString())); - return; + return false; } } }