test_pairs.py 文件源码

def test_dump_pairs_astsv(self, bitsets, number_of_bits, label2id):
        out = StringIO()

        pairs.dump_pairs(bitsets,
                         bitsets,
                         'tsv',
                         'StringIO',
                         out,
                         number_of_bits,
                         0.4,
                         0.05,
                         label2id,
                         False,
                         True,
                         )
        result = out.getvalue()

        expected = "a\tc\t0.13556\n"
        assert result == expected