Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- private class DoHloubkyIterator implements Iterator<T> {
- private Zasobnik<Prvek> zasobnik;
- private Prvek next;
- public DoHloubkyIterator() {
- zasobnik = new Zasobnik();
- next = koren;
- if (next == null) {
- throw new NullPointerException();
- }
- zasobnik.vlozPrvek(next);
- }
- @Override
- public boolean hasNext() {
- return next != null;
- }
- @Override
- public T next() {
- Prvek tmp;
- if (!hasNext()) {
- throw new NullPointerException();
- }
- tmp = zasobnik.odeberPrvek();
- if (tmp.levy != null) {
- zasobnik.vlozPrvek(tmp.levy);
- }
- if (tmp.pravy != null) {
- zasobnik.vlozPrvek(tmp.pravy);
- }
- return (T) tmp;
- }
- }
- @Test
- public void TestIterator() {
- zk.vlozKoren(to1);
- Iterator<TestClass> prohlidka = zk.iterator(eTypProhlidky.HLOUBKY);
- assertTrue(prohlidka.hasNext());
- assertEquals(to1, prohlidka.next());
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement