diff --git a/tests/test_project.py b/tests/test_project.py index 17eae3b0..7502176a 100644 --- a/tests/test_project.py +++ b/tests/test_project.py @@ -229,6 +229,11 @@ def testIndexMeta(nwMinimal, nwDummy): assert wC == 12 # Words in text and title only assert pC == 2 # Paragraphs in text only + # Look up an ivalid handle + theRefs = theIndex.getReferences("Not a handle") + assert theRefs["@pov"] == [] + assert theRefs["@char"] == [] + # The novel file should now refer to Jane as @pov and @char theRefs = theIndex.getReferences(nHandle) assert str(theRefs["@pov"]) == "['Jane']" @@ -238,6 +243,86 @@ def testIndexMeta(nwMinimal, nwDummy): theRefs = theIndex.getBackReferenceList(cHandle) assert str(theRefs) == "{'%s': 'T000001'}" % nHandle + # Get section counts for a novel file + assert theIndex.scanText(nHandle, ( + "# Hello World!\n" + "@pov: Jane\n" + "@char: Jane\n" + "\n" + "% this is a comment\n" + "\n" + "This is a story about Jane Smith.\n" + "\n" + "Well, not really.\n" + "\n" + "# Hello World!\n" + "@pov: Jane\n" + "@char: Jane\n" + "\n" + "% this is a comment\n" + "\n" + "This is a story about Jane Smith.\n" + "\n" + "Well, not really.\n" + )) + # Whole document + cC, wC, pC = theIndex.getCounts(nHandle) + assert cC == 124 + assert wC == 24 + assert pC == 4 + + # First part + cC, wC, pC = theIndex.getCounts(nHandle, "T000001") + assert cC == 62 + assert wC == 12 + assert pC == 2 + + # First part + cC, wC, pC = theIndex.getCounts(nHandle, "T000011") + assert cC == 62 + assert wC == 12 + assert pC == 2 + + # Get section counts for a note file + assert theIndex.scanText(cHandle, ( + "# Hello World!\n" + "@pov: Jane\n" + "@char: Jane\n" + "\n" + "% this is a comment\n" + "\n" + "This is a story about Jane Smith.\n" + "\n" + "Well, not really.\n" + "\n" + "# Hello World!\n" + "@pov: Jane\n" + "@char: Jane\n" + "\n" + "% this is a comment\n" + "\n" + "This is a story about Jane Smith.\n" + "\n" + "Well, not really.\n" + )) + # Whole document + cC, wC, pC = theIndex.getCounts(cHandle) + assert cC == 124 + assert wC == 24 + assert pC == 4 + + # First part + cC, wC, pC = theIndex.getCounts(cHandle, "T000001") + assert cC == 62 + assert wC == 12 + assert pC == 2 + + # First part + cC, wC, pC = theIndex.getCounts(cHandle, "T000011") + assert cC == 62 + assert wC == 12 + assert pC == 2 + assert theProject.closeProject() @pytest.mark.project