public void ApplyTest()
{
DataTable input = new DataTable("Sample data");
input.Columns.Add("Age", typeof(int));
input.Columns.Add("Classification", typeof(string));
input.Rows.Add(10, 0);
input.Rows.Add(7, 0);
input.Rows.Add(4, 0);
input.Rows.Add(21, 1);
input.Rows.Add(27, 1);
input.Rows.Add(12, 0);
input.Rows.Add(79, 0);
input.Rows.Add(40, 0);
input.Rows.Add(30, 0);
input.Rows.Add(32, 0);
{
Grouping target = new Grouping();
target.Proportion = 0.2;
DataTable actual = target.Apply(input);
DataRow[] a = actual.Select("Group = 0");
DataRow[] b = actual.Select("Group = 1");
Assert.AreEqual(2, a.Length);
Assert.AreEqual(8, b.Length);
}
{
Grouping target = new Grouping();
target.Proportion = 0.5;
DataTable actual = target.Apply(input);
DataRow[] a = actual.Select("Group = 0");
DataRow[] b = actual.Select("Group = 1");
Assert.AreEqual(5, a.Length);
Assert.AreEqual(5, b.Length);
}
{
Grouping target = new Grouping();
target.Columns.Add(new Grouping.Options("Classification"));
target.Proportion = 0.5;
DataTable actual = target.Apply(input);
DataRow[] a = actual.Select("Group = 0");
DataRow[] b = actual.Select("Group = 1");
Assert.AreEqual(5, a.Length);
Assert.AreEqual(5, b.Length);
DataRow[] a0 = actual.Select("Group = 0 AND Classification = 0");
DataRow[] a1 = actual.Select("Group = 0 AND Classification = 1");
DataRow[] b0 = actual.Select("Group = 1 AND Classification = 0");
DataRow[] b1 = actual.Select("Group = 1 AND Classification = 1");
Assert.AreEqual(4, a0.Length);
Assert.AreEqual(1, a1.Length);
Assert.AreEqual(4, b0.Length);
Assert.AreEqual(1, b1.Length);
}
}