Add more tests to NearestNeighbourWordExtractorTests

src/UglyToad.PdfPig.Tests/Dla/DlaHelper.cs

@@ -2,16 +2,23 @@
 {
     internal static class DlaHelper
     {
+        private static readonly string DlaFolder = Path.GetFullPath(Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "..", "..", "..", "Dla", "Documents"));
+        private static readonly string IntegrationFolder = Path.GetFullPath(Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "..", "..", "..", "Integration", "Documents"));
+
         public static string GetDocumentPath(string name, bool isPdf = true)
         {
-            var documentFolder = Path.GetFullPath(Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "..", "..", "..", "Dla", "Documents"));
-
             if (!name.EndsWith(".pdf") && isPdf)
             {
                 name += ".pdf";
             }
 
-            return Path.Combine(documentFolder, name);
+            string doc = Path.Combine(DlaFolder, name);
+            if (File.Exists(doc))
+            {
+                return doc;
+            }
+            
+            return Path.Combine(IntegrationFolder, name);
         }
     }
 }

src/UglyToad.PdfPig.Tests/Dla/NearestNeighbourWordExtractorTests.cs

@@ -4,19 +4,84 @@
 
     public class NearestNeighbourWordExtractorTests
     {
-        [Fact]
-        public void Words2559Doc()
+        public static IEnumerable<object[]> DataWords => new[]
         {
-            // Microsoft Word count of words = 2559
+            new object[]
+            {
+                "2559 words.pdf",
+                5118,
+                2559
+            },
+            new object[]
+            {
+                "fseprd1102849.pdf",
+                12903,
+                11177
+            },
+            new object[]
+            {
+                "90 180 270 rotated.pdf",
+                589,
+                292
+            },
+            new object[]
+            {
+                "complex rotated.pdf",
+                805,
+                403
+            },
+            new object[]
+            {
+                "no horizontal distance.pdf",
+                4,
+                2
+            },
+            new object[]
+            {
+                "no vertical distance.pdf",
+                22,
+                10
+            },
+            new object[]
+            {
+                "no vertical horizontal distance.pdf",
+                4,
+                2
+            },
+            new object[]
+            {
+                "Random 2 Columns Lists Hyph - Justified.pdf",
+                1191,
+                607
+            },
+            new object[]
+            {
+                "caly-issues-56-1.pdf",
+                184,
+                156
+            },
+            new object[]
+            {
+                "caly-issues-58-2.pdf",
+                49,
+                49
+            },
+        };
 
-            using (var document = PdfDocument.Open(DlaHelper.GetDocumentPath("2559 words.pdf")))
+        [SkippableTheory]
+        [MemberData(nameof(DataWords))]
+        public void WordCount(string path, int wordCount, int noSpacesWordCount)
+        {
+            using (var document = PdfDocument.Open(DlaHelper.GetDocumentPath(path)))
             {
                 var page = document.GetPage(1);
                 var words = NearestNeighbourWordExtractor.Instance.GetWords(page.Letters).ToArray();
 
+                Assert.Equal(wordCount, words.Length);
+
                 var noSpacesWords = words.Where(x => !string.IsNullOrEmpty(x.Text.Trim())).ToArray();
 
-                Assert.Equal(2559, noSpacesWords.Length);
+                Assert.Equal(noSpacesWordCount, noSpacesWords.Length);
             }
         }
     }

src/UglyToad.PdfPig.Tests/UglyToad.PdfPig.Tests.csproj

@@ -101,9 +101,18 @@
         <None Update="Dla\Documents\90 180 270 rotated.pdf">
             <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
         </None>
+        <None Update="Dla\Documents\caly-issues-56-1.pdf">
+          <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
+        </None>
+        <None Update="Dla\Documents\caly-issues-58-2.pdf">
+          <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
+        </None>
         <None Update="Dla\Documents\complex rotated.pdf">
             <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
         </None>
+        <None Update="Dla\Documents\fseprd1102849.pdf">
+          <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
+        </None>
         <None Update="Dla\Documents\no horizontal distance.pdf">
             <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
         </None>