def test_vocab_col_uses_text():
file_name = 'fake'
fake_text = ['fake', 'text']
col = np.random.randint(0, 5)
with patch('baseline.reader._read_from_col') as read_patch:
_ = _build_vocab_for_col(col, file_name, {}, fake_text)
read_patch.assert_not_called()
def test_vocab_col_uses_text():
file_name = 'fake'
fake_text = ['fake', 'text']
col = np.random.randint(0, 5)
with patch('baseline.reader._read_from_col') as read_patch:
_ = _build_vocab_for_col(col, file_name, {}, fake_text)
read_patch.assert_not_called()
def test_vocab_col_calls_counts():
dummy = ['text']
with patch('baseline.reader._read_from_col') as read_patch:
read_patch.return_value = dummy
vects = {'a': MagicMock(), 'b': MagicMock(), 'c': MagicMock()}
_ = _build_vocab_for_col(0, None, vects)
for vect in vects.values():
vect.count.assert_called_once_with(dummy[0])
def test_vocab_col_calls_counts():
dummy = ['text']
with patch('baseline.reader._read_from_col') as read_patch:
read_patch.return_value = dummy
vects = {'a': MagicMock(), 'b': MagicMock(), 'c': MagicMock()}
_ = _build_vocab_for_col(0, None, vects)
for vect in vects.values():
vect.count.assert_called_once_with(dummy[0])
def test_vocab_col_calls_read():
file_name = 'fake'
col = np.random.randint(0, 5)
with patch('baseline.reader._read_from_col') as read_patch:
_ = _build_vocab_for_col(col, file_name, {})
read_patch.assert_called_once_with(col, file_name)
def test_vocab_col_calls_read():
file_name = 'fake'
col = np.random.randint(0, 5)
with patch('baseline.reader._read_from_col') as read_patch:
_ = _build_vocab_for_col(col, file_name, {})
read_patch.assert_called_once_with(col, file_name, r'\t', r'\s')
