# -*- coding: utf-8 -*- # ------------------------------------------------------------------------------ # Name: corpus/testCorpus.py # Purpose: testing for the corpus # # Authors: Chris Ariza # # Copyright: Copyright © 2012 Michael Scott Asato Cuthbert # License: BSD, see license.txt # ------------------------------------------------------------------------------ from __future__ import annotations import re import unittest from music21 import common from music21 import corpus class Test(unittest.TestCase): def testGetPaths(self): for known in [ 'schumann_clara/opus17/movement3.xml', 'schoenberg/opus19/movement2.mxl', 'palestrina/agnus_02.krn', ]: a = corpus.getWork(known) # make sure it is not an empty list self.assertTrue(a) workSlashes = re.sub(r'\\', '/', str(a)) self.assertTrue(workSlashes.lower().endswith(known.lower()), (workSlashes, known)) def testBachKeys(self): from music21 import key keyObjs = [] for filePath in corpus.getComposer('bach')[23:28]: # get 5 in the middle s = corpus.parse(filePath) # get keys from first part keyObj = s.parts[0][key.KeySignature].first() keyObjs.append(keyObj) # environLocal.printDebug([keyObj]) self.assertEqual(len(keyObjs), 5) def testEssenImport(self): # can get a single file just by file name filePath = corpus.getWork('altdeu10') self.assertEqual(filePath.name, 'altdeu10.abc') self.assertEqual(filePath.parent.name, 'essenFolksong') filePathCollection = corpus.getComposer('essenFolksong') self.assertEqual(len(filePathCollection), 31) filePathCollection = corpus.getComposer('essenFolksong', fileExtensions=('abc',)) self.assertEqual(len(filePathCollection), 31) def testDesPrezImport(self): # can get a single file just by file name filePath = str(corpus.getWork('fortunaDunGranTempo')) filePath = re.sub(r'\\', '/', filePath) self.assertTrue(filePath.endswith('josquin/fortunaDunGranTempo.abc')) filePathCollection = corpus.getComposer('josquin') self.assertGreaterEqual(len(filePathCollection), 8) filePathCollection = corpus.getComposer('josquin', fileExtensions=('abc',)) self.assertGreaterEqual(len(filePathCollection), 8) # def testHandelImport(self): # # can get a single file just by file name # unused_fp = corpus.getWork('hwv56/movement1-01')# # fpCollection = corpus.getComposer('handel') # self.assertGreaterEqual(len(fpCollection), 1) # fpCollection = corpus.getComposer('handel', ['md']) # self.assertGreaterEqual(len(fpCollection), 1) def testSearch01(self): searchResults = corpus.search('china', field='locale') self.assertGreater(len(searchResults), 1200) def testSearch02(self): searchResults = corpus.search('Sichuan', field='locale') self.assertEqual(len(searchResults), 47) def testSearch03(self): searchResults = corpus.search('Taiwan', field='locale') self.assertEqual(len(searchResults), 27) pathInfo = sorted((str(searchResult.sourcePath), searchResult.number) for searchResult in searchResults) items = [ ('essenFolksong/han1.abc', '269'), ('essenFolksong/han1.abc', '270'), ('essenFolksong/han1.abc', '271'), ('essenFolksong/han1.abc', '272'), ('essenFolksong/han1.abc', '273'), ('essenFolksong/han1.abc', '274'), ('essenFolksong/han1.abc', '335'), ('essenFolksong/han1.abc', '528'), ('essenFolksong/han1.abc', '529'), ('essenFolksong/han1.abc', '530'), ('essenFolksong/han2.abc', '204'), ('essenFolksong/han2.abc', '205'), ('essenFolksong/han2.abc', '206'), ('essenFolksong/han2.abc', '207'), ('essenFolksong/han2.abc', '208'), ('essenFolksong/han2.abc', '209'), ('essenFolksong/han2.abc', '210'), ('essenFolksong/han2.abc', '211'), ('essenFolksong/han2.abc', '212'), ('essenFolksong/han2.abc', '213'), ('essenFolksong/han2.abc', '214'), ('essenFolksong/han2.abc', '215'), ('essenFolksong/han2.abc', '216'), ('essenFolksong/han2.abc', '217'), ('essenFolksong/han2.abc', '218'), ('essenFolksong/han2.abc', '219'), ('essenFolksong/han2.abc', '220'), ] if common.getPlatform() == 'win': expected = [(tup[0].replace('/', '\\'), tup[1]) for tup in items] else: expected = items self.assertEqual(pathInfo, expected) def testSearch04(self): searchResults = corpus.search('Sichuan|Taiwan', field='locale') self.assertEqual(len(searchResults), 74) def testSearch05(self): searchResults = corpus.search('bach') self.assertGreater(len(searchResults), 120) def testSearch06(self): searchResults = corpus.search('haydn', field='composer') self.assertEqual(len(searchResults), 0) searchResults = corpus.search('haydn|bach', field='composer') self.assertGreaterEqual(len(searchResults), 16) def testSearch07(self): searchResults = corpus.search('canon') self.assertGreaterEqual(len(searchResults), 1) def testSearch08(self): searchResults = corpus.search('3/8', field='timeSignature') self.assertGreater(len(searchResults), 360) def testSearch09(self): searchResults = corpus.search('3/.', field='timeSignature') self.assertGreaterEqual(len(searchResults), 2200) def testSearch10(self): from music21 import key ks = key.KeySignature(3) searchResults = corpus.search(ks, field='keySignature') self.assertEqual(len(searchResults) >= 32, True, len(searchResults)) # def testSearch12(self): # # searching virtual entries # searchResults = corpus.search('coltrane', field='composer') # self.assertGreater(len(searchResults), 0) # # returns items in pairs: url and work number # self.assertEqual(searchResults[0].sourcePath, # 'http://impromastering.com/uploads/transcription_file/' + # 'file/196/Giant_Steps__John_Coltrane_C.xml') # def testGetWorkList(self): # self.assertGreaterEqual(len(corpus.corpora.CoreCorpus().getPaths('.md')), 38) # workList = corpus.corpora.CoreCorpus().getWorkList('bach/artOfFugue_bwv1080', 1, '.zip') # self.assertEqual(len(workList), 1) # self.assertEqual(len( # corpus.corpora.CoreCorpus().getWorkList('handel/hwv56', (1, 1), '.md')), 1) # self.assertEqual(len( # corpus.corpora.CoreCorpus().getWorkList('handel/hwv56', '1-01', '.md')), 1) # self.assertEqual(len( # corpus.corpora.CoreCorpus().getWorkList('bach/artOfFugue_bwv1080')), 21) # self.assertEqual(len( # corpus.corpora.CoreCorpus().getWorkList('bach/artOfFugue_bwv1080', 1)), 1) # # # there are two versions of this file # self.assertEqual(len(corpus.getWorkList('beethoven/opus18no1', 1)), 2) # # # if specify movement # for bwv in [ # 'bwv846', 'bwv847', 'bwv848', 'bwv849', 'bwv850', 'bwv851', # 'bwv852', 'bwv853', 'bwv854', 'bwv855', 'bwv856', 'bwv857', # 'bwv858', 'bwv859', 'bwv860', 'bwv861', 'bwv862', 'bwv863', # 'bwv864', 'bwv865', 'bwv866', 'bwv867', 'bwv868', 'bwv869', # 'bwv870', 'bwv871', 'bwv872', 'bwv873', 'bwv874', 'bwv875', # 'bwv876', 'bwv877', 'bwv878', 'bwv879', 'bwv880', 'bwv881', # 'bwv882', 'bwv883', 'bwv884', 'bwv885', 'bwv886', 'bwv887', # 'bwv888', 'bwv889', 'bwv890', 'bwv891', 'bwv892', 'bwv893', # ]: # # print(bwv) # self.assertEqual(len(corpus.corpora.CoreCorpus().getWorkList(bwv)), 2) # self.assertEqual(len(corpus.corpora.CoreCorpus().getWorkList(bwv, 1)), 1) # self.assertEqual(len(corpus.corpora.CoreCorpus().getWorkList(bwv, 2)), 1) # def testWTCImport01(self): # score = corpus.parse('bach/bwv846', 1) # self.assertEqual( # score.metadata.title, # 'WTC I: Prelude and Fugue in C major', # ) # self.assertEqual(score.metadata.movementNumber, '1') # def testWTCImport02(self): # score = corpus.parse('bach/bwv846', 2) # self.assertEqual(score.metadata.movementName, 'Fugue I. ') # self.assertEqual(score.metadata.movementNumber, '2') # def testWTCImport03(self): # score = corpus.parse('bach/bwv862', 1) # self.assertEqual( # score.metadata.title, # 'WTC I: Prelude and Fugue in A flat major', # ) # def testWTCImport04(self): # score = corpus.parse('bach/bwv888', 1) # self.assertEqual( # score.metadata.title, # 'WTC II: Prelude and Fugue in A major', # ) # # s.show() # def testWorkReferences(self): # s = corpus.getWorkReferences() # # # presently 19 top level lists # self.assertGreaterEqual(len(s), 19) # self.assertEqual(len(s[0].keys()), 4) if __name__ == '__main__': import music21 music21.mainTest('noDocTest', Test)