VSinerva/markov-music-generator
Archived
1
0
Fork 0
Code Activity

Korjaa trien testit. Tehty markovin ketjun testi

Browse source
This commit is contained in:
Vili Sinervä 2022-09-22 10:46:05 +03:00
parent 7dddb59c31
commit 97a9b4fca7
2 changed files with 37 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

36
src/tests/markov_ketju_test.py Normal file
View file

@ -0,0 +1,36 @@
import unittest
from markov_ketju import MarkovKetju
class TestMarkovKetju(unittest.TestCase):
def test_seuraava_1_aste(self):
ketju = MarkovKetju(1)
opetusdata = ["ABCDEA"]
ketju.kasittele_opetusdata(opetusdata)
ketju.aseta_alkuosa("E")
self.assertEqual("A", ketju.seuraava())
self.assertEqual("B", ketju.seuraava())
self.assertEqual("C", ketju.seuraava())
self.assertEqual("D", ketju.seuraava())
self.assertEqual("E", ketju.seuraava())
self.assertEqual("A", ketju.seuraava())
ketju.aseta_alkuosa("F")
self.assertFalse(ketju.seuraava())
def test_seuraava_2_aste(self):
ketju = MarkovKetju(2)
opetusdata = ["ABCDEAB", "F"]
ketju.kasittele_opetusdata(opetusdata)
ketju.aseta_alkuosa("DE")
self.assertEqual("A", ketju.seuraava())
self.assertEqual("B", ketju.seuraava())
self.assertEqual("C", ketju.seuraava())
self.assertEqual("D", ketju.seuraava())
self.assertEqual("E", ketju.seuraava())
self.assertEqual("A", ketju.seuraava())
self.assertRaises(ValueError, ketju.aseta_alkuosa, "F")
ketju.aseta_alkuosa("EF")
self.assertFalse(ketju.seuraava())

2
src/tests/trie_test.py
View file

@ -21,7 +21,7 @@ class TestTrie(unittest.TestCase):
self.assertEqual(2, self.trie.juuri.lapset["A"].lapset["B"].laskuri)
def test_etsi_seuraavat(self):
self.assertFalse(self.trie.etsi_seuraavat("AB"))
self.assertFalse(self.trie.etsi_seuraavat("AB")[0])
self.trie.lisaa("ABC")
self.trie.lisaa("ABC")
self.assertEqual(self.trie.etsi_seuraavat("AB"), ([1], ["C"]))