def test_factor_relabel(): "Test Factor.relabel() method" f = Factor('aaabbbccc') f.update_labels({'a': 'd'}) assert_array_equal(f, Factor('dddbbbccc')) f.update_labels({'d': 'c', 'c': 'd'}) assert_array_equal(f, Factor('cccbbbddd')) f.update_labels({'d': 'c'}) assert_array_equal(f, Factor('cccbbbccc')) assert_raises(KeyError, f.update_labels, {'a':'c'})