def test_dump_pairs_badffn(self, bitsets, number_of_bits, label2id):
with pytest.raises(Exception) as cm:
pairs.dump_pairs(bitsets,
bitsets,
'hdf5_compact',
'-',
None,
number_of_bits,
0.4,
0.05,
label2id,
True
)
assert cm.value.args == ("hdf5 formats can't be outputted to stdout",)
def test_dump_pairs_badformat(self, bitsets, number_of_bits, label2id):
with pytest.raises(LookupError) as cm:
pairs.dump_pairs(bitsets,
bitsets,
'bikes',
'StringIO',
None,
number_of_bits,
0.4,
0.05,
label2id,
True
)
assert cm.value.args == ('Invalid output format',)
def test_dump_pairs_badffn(self):
with assert_raises(Exception) as cm:
pairs.dump_pairs(self.bitsets,
self.bitsets,
'hdf5_compact',
'-',
None,
self.number_of_bits,
0.4,
0.05,
self.label2id,
True
)
eq_(cm.exception.args, ("hdf5 formats can't be outputted to stdout",))
def test_dump_pairs_badformat(self):
with assert_raises(LookupError) as cm:
pairs.dump_pairs(self.bitsets,
self.bitsets,
'bikes',
'StringIO',
None,
self.number_of_bits,
0.4,
0.05,
self.label2id,
True
)
eq_(cm.exception.args, ('Invalid output format',))
def test_dump_pairs_astsv_nomem(self, bitsets, number_of_bits, label2id):
out = StringIO()
pairs.dump_pairs(bitsets,
bitsets,
'tsv',
'StringIO',
out,
number_of_bits,
0.4,
0.05,
label2id,
True,
True,
)
result = out.getvalue()
expected = "a\tc\t0.13556\n"
assert result == expected
def test_dump_pairs_astsv_nomem(self):
out = StringIO()
pairs.dump_pairs(self.bitsets,
self.bitsets,
'tsv',
'StringIO',
out,
self.number_of_bits,
0.4,
0.05,
self.label2id,
True,
True,
)
result = out.getvalue()
expected = "a\tc\t0.13556\n"
eq_(result, expected)
def test_dump_pairs_astsv_nomem(self, bitsets, number_of_bits, label2id):
out = StringIO()
pairs.dump_pairs(
bitsets,
bitsets,
'tsv',
'StringIO',
out,
number_of_bits,
0.4,
0.05,
label2id,
True,
True,
)
result = out.getvalue()
expected = {'a\tc\t0.44667', 'a\tb\t0.33333', 'b\tc\t0.77667'}
assert set(result.rstrip().split('\n')) == expected
def test_dump_pairs_badffn(self, bitsets, number_of_bits, label2id):
with pytest.raises(Exception) as cm:
pairs.dump_pairs(bitsets, bitsets, 'hdf5_compact', '-', None,
number_of_bits, 0.4, 0.05, label2id, True)
assert cm.value.args == ("hdf5 formats can't be outputted to stdout", )
def test_dump_pairs_badformat(self, bitsets, number_of_bits, label2id):
with pytest.raises(LookupError) as cm:
pairs.dump_pairs(bitsets, bitsets, 'bikes', 'StringIO', None,
number_of_bits, 0.4, 0.05, label2id, True)
assert cm.value.args == ('Invalid output format', )
