def test_categorical_merge_mapping(self): var = Categorical("C", ("a", "b1", "b2"), ()) tr = CategoriesMapping( (("a", "a"), ("b1", "b"), ("b2", "b"), (None, "c")), ) r = report_transform(var, [tr]) self.assertIn('b', r)
def test_categories_mapping(self): var = Categorical("C", ("a", "b", "c"), ()) tr = CategoriesMapping( (("a", "aa"), ("b", None), ("c", "cc"), (None, "ee")), ) r = report_transform(var, [tr]) self.assertIn("a", r) self.assertIn("aa", r) self.assertIn("b", r) self.assertIn("<s>", r)
def test_change_ordered(self): var = Categorical("C", ("a", "b"), ()) tr = ChangeOrdered(True) r = report_transform(var, [tr]) self.assertIn("ordered", r)
def test_rename(self): var = Real("X", (-1, ""), ()) tr = Rename("Y") val = report_transform(var, [tr]) self.assertIn("X", val) self.assertIn("Y", val)
def test_annotate(self): var = Real("X", (-1, ""), (("a", "1"), ("b", "z"))) tr = Annotate((("a", "2"), ("j", "z"))) r = report_transform(var, [tr]) self.assertIn("a", r) self.assertIn("b", r)
def test_reinterpret(self): var = String("T", (), False) for tr in (AsContinuous(), AsCategorical(), AsTime()): t = report_transform(var, [tr]) self.assertIn("→ (", t)
def test_categorical_merge_mapping(self): var = Categorical("C", ("a", "b1", "b2"), (), False) tr = CategoriesMapping( (("a", "a"), ("b1", "b"), ("b2", "b"), (None, "c")), ) r = report_transform(var, [tr]) self.assertIn('b', r)
def test_unlinke(self): var = Real("X", (-1, ""), (("a", "1"), ("b", "z")), True) r = report_transform(var, [Unlink()]) self.assertIn("unlinked", r)
def test_annotate(self): var = Real("X", (-1, ""), (("a", "1"), ("b", "z")), False) tr = Annotate((("a", "2"), ("j", "z"))) r = report_transform(var, [tr]) self.assertIn("a", r) self.assertIn("b", r)