def test_nonempty(self): phrase_pattern = mock.Mock() phrase_pattern.resolve.return_value = ['foo'] grammar_mock = mock.Mock() grammar_mock.get_phrase_patterns.return_value = [phrase_pattern] atom = Atom('foo', 'bar') self.assertEqual(generate_phrases(grammar_mock, [atom]), ['foo'])
def test_atoms_empty(self): phrase_pattern = mock.Mock() phrase_pattern.resolve.return_value = [] grammar_mock = mock.Mock() grammar_mock.get_phrase_patterns.return_value = [phrase_pattern] self.assertEqual(generate_phrases(grammar_mock, []), [])
def test_grammar_empty(self): grammar_mock = mock.Mock() grammar_mock.get_phrase_patterns.return_value = [] self.assertEqual(generate_phrases(grammar_mock, [Atom('foo', 'bar')]), [])