def test_transform(self): """Check if it transforms properly.""" t = Enumerate([2, 'asfa', 'ipsi']) assert t.transform(2) == 0 assert t.transform('asfa') == 1 assert t.transform('ipsi') == 2 with pytest.raises(KeyError): t.transform('aafdasfa') assert numpy.all(t.transform([['ipsi', 'asfa'], [2, 'ipsi']]) == [[2, 1], [0, 2]]) # for the crazy enough t = Enumerate([2]) assert t.transform(2) == 0 with pytest.raises(KeyError): t.transform('aafdasfa') assert numpy.all(t.transform([[2, 2], [2, 2]]) == [[0, 0], [0, 0]])
def test_transform(self): """Check if it transforms properly.""" t = Enumerate([2, "asfa", "ipsi"]) assert t.transform(2) == 0 assert t.transform("asfa") == 1 assert t.transform("ipsi") == 2 with pytest.raises(KeyError): t.transform("aafdasfa") assert numpy.all( t.transform([["ipsi", "asfa"], [2, "ipsi"]]) == [[2, 1], [0, 2]]) # for the crazy enough t = Enumerate([2]) assert t.transform(2) == 0 with pytest.raises(KeyError): t.transform("aafdasfa") assert numpy.all(t.transform([[2, 2], [2, 2]]) == [[0, 0], [0, 0]])
def test_deepcopy(self): """Verify that the transformation object can be copied""" t = Enumerate([2, 'asfa', 'ipsi']) # Copy won't fail if vectorized function is not called at least once. t.transform([2]) copy.deepcopy(t)