algorithm/Java
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Java.git synced 2025-07-27 22:43:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Enhance Trie data structure with added methods and tests (#5538)

Browse Source
This commit is contained in:
Sailok Chinta
2024-10-08 02:41:55 +05:30
committed by GitHub
parent 6868bf8ba0
commit 5dcf6c0f29
3 changed files with 129 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
src/test/java/com/thealgorithms/datastructures/trees/TrieImpTest.java → src/test/java/com/thealgorithms/datastructures/trees/TrieTest.java
View File

@ -1,17 +1,21 @@
package com.thealgorithms.datastructures.trees;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertTrue;
import java.util.List;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
public class TrieImpTest {
private TrieImp trie;
public class TrieTest {
private static final List<String> WORDS = List.of("Apple", "App", "app", "APPLE");
private Trie trie;
@BeforeEach
public void setUp() {
trie = new TrieImp();
trie = new Trie();
}
@Test
@ -66,11 +70,26 @@ public class TrieImpTest {
}
@Test
public void testIsValidWord() {
assertTrue(TrieImp.isValid("validword"), "Word should be valid (only lowercase letters).");
assertFalse(TrieImp.isValid("InvalidWord"), "Word should be invalid (contains uppercase letters).");
assertFalse(TrieImp.isValid("123abc"), "Word should be invalid (contains numbers).");
assertFalse(TrieImp.isValid("hello!"), "Word should be invalid (contains special characters).");
assertFalse(TrieImp.isValid(""), "Empty string should be invalid.");
public void testCountWords() {
Trie trie = createTrie();
assertEquals(WORDS.size(), trie.countWords(), "Count words should return the correct number of words.");
}
@Test
public void testStartsWithPrefix() {
Trie trie = createTrie();
assertTrue(trie.startsWithPrefix("App"), "Starts with prefix should return true.");
}
@Test
public void testCountWordsWithPrefix() {
Trie trie = createTrie();
assertEquals(2, trie.countWordsWithPrefix("App"), "Count words with prefix should return 2.");
}
private Trie createTrie() {
Trie trie = new Trie();
WORDS.forEach(trie::insert);
return trie;
}
}