deepword.tests package¶
Submodules¶
deepword.tests.test_agent_input module¶
-
class
deepword.tests.test_agent_input.TestAgentInput(methodName='runTest')¶ Bases:
unittest.case.TestCase-
test_align_batch_str()¶
-
test_bert_commonsense_input()¶
-
test_dqn_input()¶
-
test_drrn_action_input()¶
-
test_sample_ids()¶
-
-
deepword.tests.test_agent_input.gen_action_collector(hp: tensorflow.contrib.training.python.training.hparam.HParams, tokenizer: deepword.tokenizers.Tokenizer) → deepword.action.ActionCollector¶
-
deepword.tests.test_agent_input.gen_action_master(vocab: List[str], turns_up_to: int, n_rows: int) → List[List[deepword.agents.utils.ActionMaster]]¶
-
deepword.tests.test_agent_input.gen_rand_str(vocab: List[str], length_up_to: int, n_rows: int, allow_empty_str: bool = True) → List[str]¶