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_rename(self):
var = Real("X", (-1, ""), ())
tr = Rename("Y")
val = report_transform(var, [tr])
self.assertIn("X", val)
self.assertIn("Y", val)
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)
