-
Notifications
You must be signed in to change notification settings - Fork 587
/
PartitionTreeBagTest.java
34 lines (31 loc) · 1.52 KB
/
PartitionTreeBagTest.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
/*
* Copyright (c) 2022 The Bank of New York Mellon.
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* and Eclipse Distribution License v. 1.0 which accompany this distribution.
* The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html
* and the Eclipse Distribution License is available at
* http://www.eclipse.org/org/documents/edl-v10.php.
*/
package org.eclipse.collections.impl.partition.bag.sorted;
import org.eclipse.collections.api.bag.sorted.MutableSortedBag;
import org.eclipse.collections.api.factory.SortedBags;
import org.eclipse.collections.api.partition.bag.sorted.PartitionImmutableSortedBag;
import org.eclipse.collections.impl.block.factory.Comparators;
import org.junit.Assert;
import org.junit.Test;
public class PartitionTreeBagTest
{
@Test
public void toImmutable()
{
PartitionTreeBag<Integer> partitionTreeBag = new PartitionTreeBag<>(Comparators.naturalOrder());
MutableSortedBag<Integer> selected = SortedBags.mutable.of(1, 2, 3);
MutableSortedBag<Integer> rejected = SortedBags.mutable.of(4, 5, 6);
partitionTreeBag.getSelected().addAll(selected);
partitionTreeBag.getRejected().addAll(rejected);
PartitionImmutableSortedBag<Integer> immutableSortedBag = partitionTreeBag.toImmutable();
Assert.assertEquals(selected, immutableSortedBag.getSelected());
Assert.assertEquals(rejected, immutableSortedBag.getRejected());
}
}