Search in sources :

Example 1 with Words

use of com.ibm.watson.developer_cloud.speech_to_text.v1.model.Words in project java-sdk by watson-developer-cloud.

the class SpeechToTextIT method testListWordsTypeAll.

/**
 * Test list words with type all.
 */
@Test
@Ignore
public void testListWordsTypeAll() {
    ListWordsOptions listOptions = new ListWordsOptions.Builder().customizationId(customizationId).wordType(ListWordsOptions.WordType.ALL).build();
    Words result = service.listWords(listOptions).execute();
    assertNotNull(result);
    assertTrue(!result.getWords().isEmpty());
}
Also used : Words(com.ibm.watson.developer_cloud.speech_to_text.v1.model.Words) ListWordsOptions(com.ibm.watson.developer_cloud.speech_to_text.v1.model.ListWordsOptions) Ignore(org.junit.Ignore) WatsonServiceTest(com.ibm.watson.developer_cloud.WatsonServiceTest) Test(org.junit.Test)

Example 2 with Words

use of com.ibm.watson.developer_cloud.speech_to_text.v1.model.Words in project java-sdk by watson-developer-cloud.

the class SpeechToTextIT method testCreateLanguageModel.

/**
 * Test create language model.
 *
 * @throws InterruptedException the interrupted exception
 */
@Test
public void testCreateLanguageModel() throws InterruptedException, FileNotFoundException {
    CreateLanguageModelOptions createOptions = new CreateLanguageModelOptions.Builder().name("java-sdk-temporary").baseModelName(EN_BROADBAND16K).description("Temporary custom model for testing the Java SDK").build();
    LanguageModel myModel = service.createLanguageModel(createOptions).execute();
    String id = myModel.getCustomizationId();
    try {
        // Add a corpus file to the model
        AddCorpusOptions addOptions = new AddCorpusOptions.Builder().customizationId(id).corpusName("corpus-1").corpusFile(new File(String.format(SPEECH_RESOURCE, "corpus1.txt"))).corpusFileContentType(HttpMediaType.TEXT_PLAIN).allowOverwrite(false).build();
        service.addCorpus(addOptions).execute();
        // Get corpus status
        GetCorpusOptions getOptions = new GetCorpusOptions.Builder().customizationId(id).corpusName("corpus-1").build();
        for (int x = 0; x < 30 && !service.getCorpus(getOptions).execute().getStatus().equals(Status.ANALYZED); x++) {
            Thread.sleep(5000);
        }
        assertTrue(service.getCorpus(getOptions).execute().getStatus().equals(Status.ANALYZED));
        // Add the corpus file to the model again and allow overwrite
        AddCorpusOptions addOptionsWithOverwrite = new AddCorpusOptions.Builder().customizationId(id).corpusName("corpus-1").corpusFile(new File(String.format(SPEECH_RESOURCE, "corpus1.txt"))).corpusFileContentType(HttpMediaType.TEXT_PLAIN).allowOverwrite(true).build();
        service.addCorpus(addOptionsWithOverwrite).execute();
        // Get corpus status
        for (int x = 0; x < 30 && !service.getCorpus(getOptions).execute().getStatus().equals(Status.ANALYZED); x++) {
            Thread.sleep(5000);
        }
        assertTrue(service.getCorpus(getOptions).execute().getStatus().equals(Status.ANALYZED));
        // Get corpora
        ListCorporaOptions listCorporaOptions = new ListCorporaOptions.Builder().customizationId(id).build();
        Corpora corpora = service.listCorpora(listCorporaOptions).execute();
        assertNotNull(corpora);
        assertTrue(corpora.getCorpora().size() == 1);
        // Now add some user words to the custom model
        service.addWord(new AddWordOptions.Builder().customizationId(id).wordName("IEEE").word("IEEE").displayAs("IEEE").addSoundsLike("I. triple E.").build()).execute();
        service.addWord(new AddWordOptions.Builder().customizationId(id).wordName("hhonors").word("hhonors").displayAs("IEEE").addSoundsLike("H. honors").addSoundsLike("Hilton honors").build()).execute();
        service.addWord(new AddWordOptions.Builder().customizationId(id).wordName("aaa").word("aaa").displayAs("aaa").addSoundsLike("aaa").addSoundsLike("bbb").build()).execute();
        service.addWord(new AddWordOptions.Builder().customizationId(id).wordName("bbb").word("bbb").addSoundsLike("aaa").addSoundsLike("bbb").build()).execute();
        service.addWord(new AddWordOptions.Builder().customizationId(id).wordName("ccc").word("ccc").displayAs("ccc").build()).execute();
        service.addWord(new AddWordOptions.Builder().customizationId(id).wordName("ddd").word("ddd").build()).execute();
        service.addWord(new AddWordOptions.Builder().customizationId(id).wordName("eee").word("eee").build()).execute();
        // Display all words in the words resource (coming from OOVs from the corpus add and the new words just added)
        ListWordsOptions listWordsOptions = new ListWordsOptions.Builder().customizationId(id).wordType(ListWordsOptions.WordType.ALL).build();
        Words words = service.listWords(listWordsOptions).execute();
        assertNotNull(words);
    } finally {
        DeleteLanguageModelOptions deleteOptions = new DeleteLanguageModelOptions.Builder().customizationId(id).build();
        service.deleteLanguageModel(deleteOptions).execute();
    }
}
Also used : CreateLanguageModelOptions(com.ibm.watson.developer_cloud.speech_to_text.v1.model.CreateLanguageModelOptions) AddCorpusOptions(com.ibm.watson.developer_cloud.speech_to_text.v1.model.AddCorpusOptions) DeleteLanguageModelOptions(com.ibm.watson.developer_cloud.speech_to_text.v1.model.DeleteLanguageModelOptions) AddWordOptions(com.ibm.watson.developer_cloud.speech_to_text.v1.model.AddWordOptions) Corpora(com.ibm.watson.developer_cloud.speech_to_text.v1.model.Corpora) Words(com.ibm.watson.developer_cloud.speech_to_text.v1.model.Words) ListWordsOptions(com.ibm.watson.developer_cloud.speech_to_text.v1.model.ListWordsOptions) LanguageModel(com.ibm.watson.developer_cloud.speech_to_text.v1.model.LanguageModel) GetCorpusOptions(com.ibm.watson.developer_cloud.speech_to_text.v1.model.GetCorpusOptions) File(java.io.File) ListCorporaOptions(com.ibm.watson.developer_cloud.speech_to_text.v1.model.ListCorporaOptions) WatsonServiceTest(com.ibm.watson.developer_cloud.WatsonServiceTest) Test(org.junit.Test)

Example 3 with Words

use of com.ibm.watson.developer_cloud.speech_to_text.v1.model.Words in project java-sdk by watson-developer-cloud.

the class SpeechToTextIT method testListWordsIdAndType.

/**
 * Test list words with a customization ID and word type.
 */
@Test
@Ignore
public void testListWordsIdAndType() {
    ListWordsOptions listOptions = new ListWordsOptions.Builder().customizationId(customizationId).wordType(ListWordsOptions.WordType.CORPORA).build();
    Words result = service.listWords(listOptions).execute();
    assertNotNull(result);
    assertTrue(!result.getWords().isEmpty());
}
Also used : Words(com.ibm.watson.developer_cloud.speech_to_text.v1.model.Words) ListWordsOptions(com.ibm.watson.developer_cloud.speech_to_text.v1.model.ListWordsOptions) Ignore(org.junit.Ignore) WatsonServiceTest(com.ibm.watson.developer_cloud.WatsonServiceTest) Test(org.junit.Test)

Example 4 with Words

use of com.ibm.watson.developer_cloud.speech_to_text.v1.model.Words in project java-sdk by watson-developer-cloud.

the class SpeechToTextTest method testAddWords.

/**
 * Test add words.
 *
 * @throws InterruptedException the interrupted exception
 * @throws FileNotFoundException the file not found exception
 */
@Test
public void testAddWords() throws InterruptedException, FileNotFoundException {
    String id = "foo";
    Word newWord = loadFixture("src/test/resources/speech_to_text/word.json", Word.class);
    Map<String, Object> wordsAsMap = new HashMap<String, Object>();
    wordsAsMap.put("words", new Word[] { newWord });
    server.enqueue(new MockResponse().addHeader(CONTENT_TYPE, HttpMediaType.APPLICATION_JSON).setBody("{}"));
    CustomWord word = new CustomWord();
    word.setWord(newWord.getWord());
    word.setDisplayAs(newWord.getDisplayAs());
    word.setSoundsLike(newWord.getSoundsLike());
    AddWordsOptions addOptions = new AddWordsOptions.Builder().customizationId(id).addWords(word).build();
    service.addWords(addOptions).execute();
    final RecordedRequest request = server.takeRequest();
    assertEquals("POST", request.getMethod());
    assertEquals(String.format(PATH_WORDS, id), request.getPath());
    assertEquals(GSON.toJson(wordsAsMap), request.getBody().readUtf8());
}
Also used : RecordedRequest(okhttp3.mockwebserver.RecordedRequest) MockResponse(okhttp3.mockwebserver.MockResponse) CustomWord(com.ibm.watson.developer_cloud.speech_to_text.v1.model.CustomWord) Word(com.ibm.watson.developer_cloud.speech_to_text.v1.model.Word) AddWordsOptions(com.ibm.watson.developer_cloud.speech_to_text.v1.model.AddWordsOptions) HashMap(java.util.HashMap) CustomWord(com.ibm.watson.developer_cloud.speech_to_text.v1.model.CustomWord) JsonObject(com.google.gson.JsonObject) ByteString(okio.ByteString) WatsonServiceUnitTest(com.ibm.watson.developer_cloud.WatsonServiceUnitTest) Test(org.junit.Test)

Example 5 with Words

use of com.ibm.watson.developer_cloud.speech_to_text.v1.model.Words in project java-sdk by watson-developer-cloud.

the class SpeechToTextTest method testListWordsType.

/**
 * Test list words with word type all.
 *
 * @throws InterruptedException the interrupted exception
 * @throws FileNotFoundException the file not found exception
 */
@Test
public void testListWordsType() throws InterruptedException, FileNotFoundException {
    String id = "foo";
    String wordsAsStr = getStringFromInputStream(new FileInputStream("src/test/resources/speech_to_text/words.json"));
    JsonObject words = new JsonParser().parse(wordsAsStr).getAsJsonObject();
    server.enqueue(new MockResponse().addHeader(CONTENT_TYPE, HttpMediaType.APPLICATION_JSON).setBody(wordsAsStr));
    ListWordsOptions listOptions = new ListWordsOptions.Builder().customizationId(id).wordType(ListWordsOptions.WordType.ALL).build();
    Words result = service.listWords(listOptions).execute();
    final RecordedRequest request = server.takeRequest();
    assertEquals("GET", request.getMethod());
    assertEquals(String.format(PATH_WORDS, id) + "?word_type=all", request.getPath());
    assertEquals(words.get("words"), GSON.toJsonTree(result.getWords()));
}
Also used : RecordedRequest(okhttp3.mockwebserver.RecordedRequest) MockResponse(okhttp3.mockwebserver.MockResponse) Words(com.ibm.watson.developer_cloud.speech_to_text.v1.model.Words) ListWordsOptions(com.ibm.watson.developer_cloud.speech_to_text.v1.model.ListWordsOptions) JsonObject(com.google.gson.JsonObject) ByteString(okio.ByteString) FileInputStream(java.io.FileInputStream) JsonParser(com.google.gson.JsonParser) WatsonServiceUnitTest(com.ibm.watson.developer_cloud.WatsonServiceUnitTest) Test(org.junit.Test)

Aggregations

Test (org.junit.Test)16 Words (com.ibm.watson.developer_cloud.speech_to_text.v1.model.Words)12 ListWordsOptions (com.ibm.watson.developer_cloud.speech_to_text.v1.model.ListWordsOptions)11 WatsonServiceTest (com.ibm.watson.developer_cloud.WatsonServiceTest)10 Words (com.ibm.watson.developer_cloud.text_to_speech.v1.model.Words)7 WatsonServiceUnitTest (com.ibm.watson.developer_cloud.WatsonServiceUnitTest)6 ListWordsOptions (com.ibm.watson.developer_cloud.text_to_speech.v1.model.ListWordsOptions)6 Word (com.ibm.watson.developer_cloud.text_to_speech.v1.model.Word)6 RecordedRequest (okhttp3.mockwebserver.RecordedRequest)6 JsonObject (com.google.gson.JsonObject)5 RequestBuilder (com.ibm.watson.developer_cloud.http.RequestBuilder)5 MockResponse (okhttp3.mockwebserver.MockResponse)5 ByteString (okio.ByteString)5 Ignore (org.junit.Ignore)5 JsonParser (com.google.gson.JsonParser)4 FileInputStream (java.io.FileInputStream)4 Corpora (com.ibm.watson.developer_cloud.speech_to_text.v1.model.Corpora)3 AddWordOptions (com.ibm.watson.developer_cloud.text_to_speech.v1.model.AddWordOptions)3 AddWordsOptions (com.ibm.watson.developer_cloud.text_to_speech.v1.model.AddWordsOptions)3 File (java.io.File)3