public void BadLinesHandlingTest()
{
var junk = @"blablabla";
var ccbLine = @"{ ""Timestamp"": ""2019-08-27T12:45:53.6300000Z"", ""Version"": ""1"", ""c"": { ""GUser"": { ""shared_feature"": ""feature"" }, ""_multi"": [ { ""TAction"": { ""feature1"": 3.0, ""feature2"": ""name1"" } }, { ""TAction"": { ""feature1"": 3.0, ""feature2"": ""name1"" } }, { ""TAction"": { ""feature1"": 3.0, ""feature2"": ""name1"" } } ], ""_slots"": [ { ""size"": ""small"", ""_inc"": [0, 2] }, { ""size"": ""large"" } ] }, ""_outcomes"": [ { ""_id"": ""62ddd79e-4d75-4c64-94f1-a5e13a75c2e4"", ""_label_cost"": 0, ""_a"": [2, 0], ""_p"": [0.9, 0.1], ""_o"": [] }, { ""_id"": ""042661c4-d433-4b05-83d6-d51a2d1c68be"", ""_label_cost"": 0, ""_a"": [1, 0], ""_p"": [0.1, 0.9], ""_o"": [-1.0, 0.0] } ], ""VWState"": { ""m"": ""da63c529-018b-44b1-ad0f-c2b13056832c/195fc8ed-224f-471a-90c4-d3e60b336f8f"" } }";
var extractor = new CcbExtractor();
var output = new USqlUpdatableRow(CreateDefaultRow(CreateErrorHandlingSchema()));
using (var stream = new MemoryStream(Encoding.UTF8.GetBytes($"{junk}\n{ccbLine}")))
{
var input = new USqlStreamReader(stream);
int counter = 0;
foreach (var outputRow in extractor.Extract(input, output))
{
if (counter++ == 0)
{
Assert.IsFalse(string.IsNullOrWhiteSpace(output.Get <string>("ParseError")));
}
else
{
Assert.IsNotNull(output.Get <string>("SessionId"));
Assert.IsTrue(string.IsNullOrWhiteSpace(output.Get <string>("ParseError")));
}
}
}
}
public void CcbTest()
{
var ccbLine = @"{ ""Timestamp"": ""2019-08-27T12:45:53.0000000Z"", ""Version"": ""1"", ""c"": { ""GUser"": { ""shared_feature"": ""feature"" }, ""_multi"": [ { ""TAction"": { ""feature1"": 3.0, ""feature2"": ""name1"" } }, { ""TAction"": { ""feature1"": 3.0, ""feature2"": ""name1"" } }, { ""TAction"": { ""feature1"": 3.0, ""feature2"": ""name1"" } } ], ""_slots"": [ { ""size"": ""small"", ""_inc"": [0, 2] }, { ""size"": ""large"" } ] }, ""_outcomes"": [ { ""_label_cost"": 0, ""_id"": ""62ddd79e-4d75-4c64-94f1-a5e13a75c2e4"", ""_a"": [2, 0], ""_p"": [0.9, 0.1], ""_o"": [] }, { ""_label_cost"": -1.0, ""_id"": ""042661c4-d433-4b05-83d6-d51a2d1c68be"", ""_a"": [1, 0], ""_p"": [0.1, 0.9], ""_o"": [{""v"": 1.0}] } ], ""VWState"": { ""m"": ""da63c529-018b-44b1-ad0f-c2b13056832c/195fc8ed-224f-471a-90c4-d3e60b336f8f"" } }";
var extractor = new CcbExtractor();
var output = new USqlUpdatableRow(CreateDefaultRow(CreateCcbBasicSchema()));
using (var stream = new MemoryStream(Encoding.UTF8.GetBytes($"{ccbLine}")))
{
var input = new USqlStreamReader(stream);
int counter = 0;
foreach (var outputRow in extractor.Extract(input, output))
{
if (counter == 0)
{
Assert.IsNotNull(output.Get <string>("SessionId"));
Assert.AreEqual(0, output.Get <int>("SlotIdx"));
Assert.AreEqual("62ddd79e-4d75-4c64-94f1-a5e13a75c2e4", output.Get <string>("EventId"));
Assert.AreEqual(new DateTime(2019, 8, 27, 12, 45, 53, 0), output.Get <DateTime>("Timestamp"));
Assert.AreEqual(0.0, output.Get <float>("Cost"), 1e-6);
Assert.AreEqual(0.9, output.Get <float>("Prob"), 1e-6);
Assert.AreEqual(2, output.Get <int>("NumActionsPerSlot"));
Assert.AreEqual(3, output.Get <int>("NumActions"));
Assert.AreEqual(2, output.Get <int>("Action"));
Assert.AreEqual(1, output.Get <int>("CbAction"));
Assert.AreEqual(0, output.Get <int>("HasObservations"));
Assert.AreEqual(0, output.Get <float>("pdrop"));
Assert.AreEqual(false, output.Get <bool>("IsDangling"));
}
else if (counter == 1)
{
Assert.IsNotNull(output.Get <string>("SessionId"));
Assert.AreEqual(1, output.Get <int>("SlotIdx"));
Assert.AreEqual("042661c4-d433-4b05-83d6-d51a2d1c68be", output.Get <string>("EventId"));
Assert.AreEqual(new DateTime(2019, 8, 27, 12, 45, 53, 0), output.Get <DateTime>("Timestamp"));
Assert.AreEqual(-1.0, output.Get <float>("Cost"), 1e-6);
Assert.AreEqual(0.1, output.Get <float>("Prob"), 1e-6);
Assert.AreEqual(2, output.Get <int>("NumActionsPerSlot"));
Assert.AreEqual(3, output.Get <int>("NumActions"));
Assert.AreEqual(1, output.Get <int>("Action"));
Assert.AreEqual(1, output.Get <int>("CbAction"));
Assert.AreEqual(1, output.Get <int>("HasObservations"));
Assert.AreEqual(0, output.Get <float>("pdrop"));
Assert.AreEqual(false, output.Get <bool>("IsDangling"));
}
counter++;
}
}
}
public void CcbDanglingRewardTest()
{
var danglingReward = @"{""RewardValue"":846.7236,""ActionTaken"":false,""EnqueuedTimeUtc"":""2018-12-13T03:27:57.000Z"",""EventId"":""id"",""Observations"":[{""v"":846.7236,""ActionTaken"":false,""EventId"":""id"",""ActionId"":null}]}";
var extractor = new CcbExtractor();
var output = new USqlUpdatableRow(CreateDefaultRow(CreateDanglingRewardSchema()));
using (var stream = new MemoryStream(Encoding.UTF8.GetBytes(danglingReward)))
{
var input = new USqlStreamReader(stream);
foreach (var outputRow in extractor.Extract(input, output))
{
Assert.AreEqual("id", output.Get <string>("EventId"));
Assert.AreEqual(new DateTime(2018, 12, 13, 3, 27, 57), output.Get <DateTime>("EnqueuedTimeUtc"));
}
}
}
