def test_interval_indexer_1d(self): # BWV7.7: first 26 things in soprano part (simple; no quality) test_parts = [self.bwv77_soprano] expected = TestHorizIntervalIndexerLong.bwv77_S_B_short_noqual['a'] setts = {u'simple or compound': u'simple', u'quality': False} int_indexer = HorizontalIntervalIndexer(test_parts, setts) actual = int_indexer.run()[u'interval.HorizontalIntervalIndexer']['0'].iloc[:26] self.assertSequenceEqual(list(expected.index), list(actual.index)) self.assertSequenceEqual(list(expected), list(actual))
def test_interval_indexer_1c(self): # BWV7.7: first 26 things in soprano part (simple; quality) test_parts = pandas.concat([self.bwv77_soprano], axis=1) expected = TestHorizIntervalIndexerLong.bwv77_S_B_short['a'] setts = {'simple or compound': 'simple', 'quality': True} int_indexer = HorizontalIntervalIndexer(test_parts, setts) actual = int_indexer.run()['interval.HorizontalIntervalIndexer']['0'].iloc[:26] self.assertSequenceEqual(list(expected.index), list(actual.index)) self.assertSequenceEqual(list(expected), list(actual))
def test_interval_indexer_2(self): # BWV7.7: whole soprano part test_parts = [self.bwv77_soprano] expected = TestHorizIntervalIndexerLong.bwv77_S_B_basis setts = {u'simple or compound': u'compound', u'quality': True} int_indexer = HorizontalIntervalIndexer(test_parts, setts) res = int_indexer.run() actual = res[0] self.assertEqual(len(expected), len(actual)) for i, ind in enumerate(list(actual.index)): self.assertEqual(expected[i][0], ind) self.assertEqual(expected[i][1], actual[ind])
def test_interval_indexer_1d(self): # BWV7.7: first 26 things in soprano part (simple; no quality) test_parts = [self.bwv77_soprano] expected = TestHorizIntervalIndexerLong.bwv77_S_B_short_noqual setts = {u'simple or compound': u'simple', u'quality': False} int_indexer = HorizontalIntervalIndexer(test_parts, setts) res = int_indexer.run() actual = res[0].iloc[:26] self.assertEqual(len(expected), len(actual)) for i, ind in enumerate(list(actual.index)): self.assertEqual(expected[i][0], ind) self.assertEqual(expected[i][1], actual[ind])
def test_interval_indexer_2(self): # BWV7.7: whole soprano part # NB: this test is more rigourous than the others, since it actually uses the DataFrame test_parts = [self.bwv77_soprano] expected = {'0': TestHorizIntervalIndexerLong.bwv77_S_B_basis['a']} setts = {u'simple or compound': u'compound', u'quality': True} int_indexer = HorizontalIntervalIndexer(test_parts, setts) actual = int_indexer.run()[u'interval.HorizontalIntervalIndexer'] self.assertEqual(len(expected), len(actual.columns)) for key in expected.iterkeys(): self.assertTrue(key in actual) self.assertSequenceEqual(list(expected[key].index), list(actual[key].index)) self.assertSequenceEqual(list(expected[key]), list(actual[key]))