From 1240e488d1a7e37708b47df5e2728b5c8f9db09d Mon Sep 17 00:00:00 2001
From: Tim Vaughan <vaughant@bs-mbpr538.d.ethz.ch>
Date: Tue, 5 Oct 2021 17:41:45 +0200
Subject: [PATCH] Improved unit test.

---
 test/bacter/util/BacterACGLogReaderTest.java | 14 +++-----------
 1 file changed, 3 insertions(+), 11 deletions(-)

diff --git a/test/bacter/util/BacterACGLogReaderTest.java b/test/bacter/util/BacterACGLogReaderTest.java
index 88621be..7c49a22 100644
--- a/test/bacter/util/BacterACGLogReaderTest.java
+++ b/test/bacter/util/BacterACGLogReaderTest.java
@@ -3,27 +3,18 @@
 import bacter.ConversionGraph;
 import beast.util.Randomizer;
 import beast.util.XMLParser;
-import beast.util.XMLParserException;
 import org.junit.Test;
-import org.xml.sax.SAXException;
-import test.beast.beast2vs1.trace.Expectation;
-import test.beast.beast2vs1.trace.LogAnalyser;
 
-import javax.xml.parsers.ParserConfigurationException;
 import java.io.File;
-import java.io.IOException;
 import java.nio.file.Files;
 import java.nio.file.Paths;
-import java.util.ArrayList;
-import java.util.List;
 
-import static org.junit.Assert.assertTrue;
+import static org.junit.Assert.assertEquals;
 
 public class BacterACGLogReaderTest {
 
     @Test
     public void test() throws Exception {
-
         Randomizer.setSeed(1);
 
         XMLParser parser = new XMLParser();
@@ -33,8 +24,9 @@ public void test() throws Exception {
 
         BacterACGLogReader logReader = new BacterACGLogReader(new File("simulateACGs2taxon.trees"), 0.0);
 
-        System.out.println("ACG count: " + logReader.getACGCount());
+        assertEquals(100, logReader.getACGCount());
 
+        // The following simply tests whether the ACGs can be loaded without error:
         System.out.println("Root ages:");
         for (ConversionGraph acg : logReader)
             System.out.println(acg.getRoot().getHeight());