Sean/ARM64
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Refactor unit tests for CNNGenerateOutputFileBollinger and CNNGenerateOutputFileBollingerWithVIX

Browse Source
This commit is contained in:
Sean
2026-03-11 11:21:25 -04:00
parent 8715bed4cf
commit 2923971656
3 changed files with 44 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
MarketDataUnitTests/MarketDataUnitTestClass.cs
View File

@@ -186,11 +186,16 @@ public class MarketDataUnitTestClass
Assert.IsTrue(result);
}
/// <summary>
/// CNNGenerateOutputFileBollingerWithVIX - This generates the new bollinger band with VIX 224x224 image with 90 days of history
/// Note that output goes in "/home/pi/ARM64/MarketDataUnitTests/TestResults"
/// </summary>
[TestMethod]
public void CNNGenerateOutputFileVIX()
public void CNNGenerateOutputFileBollingerWithVIX()
{
DataProcessor dataProcessor=new DataProcessor();
int imageDimensions=224;
int dayCount = 90;
dataProcessor.Width=imageDimensions;
dataProcessor.Height=imageDimensions;
dataProcessor.PenWidth=1;
@@ -201,21 +206,27 @@ public class MarketDataUnitTestClass
cmParams.UseCNN = true;
cmParams.UseCNNHost = "http://" + cnnHostName + ":5000";
cmParams.UseCNNDayCount = 90;
cmParams.UseCNNDayCount = dayCount;
cmParams.UseCNNRewardPercentDecimal = 0.25;
cmCandidate.Symbol = "MIDD";
cmCandidate.TradeDate = DateTime.Parse("07-01-2024");
TestCase testCase=new TestCase(cmCandidate.Symbol,cmCandidate.TradeDate,cmParams.UseCNNDayCount,TestCase.CaseType.Test,TestCase.GenerateType.BollingerBandWithVIX,TestCase.OutputType.OutputFile);
testCase.HomePath="/home/pi/ARM64/MarketDataUnitTests/TestResults";
dataProcessor.ProcessData(testCase);
}
/// <summary>
/// CNNGenerateOutputFileBollinger - This generates the legacy bollinger band 128x128 image with 270 days of history
/// Note that output goes in "/home/pi/ARM64/MarketDataUnitTests/TestResults"
/// </summary>
[TestMethod]
public void CNNGenerateOutputFileBand()
public void CNNGenerateOutputFileBollinger()
{
DataProcessor dataProcessor=new DataProcessor();
int imageDimensions=224;
int imageDimensions=128;
int dayCount=270;
dataProcessor.Width=imageDimensions;
dataProcessor.Height=imageDimensions;
dataProcessor.PenWidth=1;
@@ -226,18 +237,18 @@ public class MarketDataUnitTestClass
cmParams.UseCNN = true;
cmParams.UseCNNHost = "http://" + cnnHostName + ":5000";
cmParams.UseCNNDayCount = 90;
cmParams.UseCNNDayCount = dayCount;
cmParams.UseCNNRewardPercentDecimal = 0.25;
cmCandidate.Symbol = "MIDD";
cmCandidate.TradeDate = DateTime.Parse("07-01-2024");
TestCase testCase=new TestCase(cmCandidate.Symbol,cmCandidate.TradeDate,cmParams.UseCNNDayCount,TestCase.CaseType.Test,TestCase.GenerateType.BollingerBand,TestCase.OutputType.OutputFile);
testCase.HomePath="/home/pi/ARM64/MarketDataUnitTests/TestResults";
dataProcessor.ProcessData(testCase);
}
[TestMethod]
public void PremarketRetrieval()
{