public void test_keep_attributes_by_names()
{
Runtime rt = TestingHelpers.LoadSmallRuntime <TitanicDataRow>("titanic_train.csv", 0, 3);
AttributesRemover ar = new AttributesRemover(rt);
string[] initial = rt.EnumerateAttributes.Select(a => a.Name).ToArray();
ar.KeepAttributes("survived", "embarked", "sex");
string[] after = rt.EnumerateAttributes.Select(a => a.Name).ToArray();
Assert.AreEqual(new[] { "survived", "pclass", "sex", "embarked" }, initial);
Assert.AreEqual(new[] { "survived", "sex", "embarked" }, after);
}
[Test] public void test_removing_attributes_by_index()
{
var rt = TestingHelpers.LoadSmallRuntime <TitanicDataRow>("titanic_train.csv", 0, 3);
var ar = new AttributesRemover(rt);
var initial = rt.EnumerateAttributes.Select(a => a.Name).ToArray();
ar.RemoveAttributes(2, 3);
var after = rt.EnumerateAttributes.Select(a => a.Name).ToArray();
Assert.AreEqual(new [] { "survived", "pclass", "sex", "embarked" }, initial);
Assert.AreEqual(new [] { "survived", "pclass" }, after);
}
