add a basic learn/retrieve test since I broke it a while back
This commit is contained in:
parent
0227b74eee
commit
667a85aa46
19
tests/fixtures/markov_fixture.json
vendored
Normal file
19
tests/fixtures/markov_fixture.json
vendored
Normal file
@ -0,0 +1,19 @@
|
||||
[
|
||||
{
|
||||
"model": "markov.markovcontext",
|
||||
"pk": 1,
|
||||
"fields": {
|
||||
"name": "#factory"
|
||||
}
|
||||
},
|
||||
{
|
||||
"model": "markov.markovtarget",
|
||||
"pk": 1,
|
||||
"fields": {
|
||||
"name": "#factory",
|
||||
"context": 1,
|
||||
"channel": 1,
|
||||
"chatter_chance": 0
|
||||
}
|
||||
}
|
||||
]
|
17
tests/test_markov_lib.py
Normal file
17
tests/test_markov_lib.py
Normal file
@ -0,0 +1,17 @@
|
||||
"""Test markov utility methods."""
|
||||
from django.test import TestCase
|
||||
|
||||
from markov.lib import get_word_out_of_states, learn_line
|
||||
from markov.models import MarkovContext, MarkovState
|
||||
|
||||
|
||||
class MarkovLibTestCase(TestCase):
|
||||
"""Test library methods used by the Markov plugin."""
|
||||
|
||||
fixtures = ['tests/fixtures/irc_server_fixture.json', 'tests/fixtures/markov_fixture.json']
|
||||
|
||||
def test_learn_and_get(self):
|
||||
"""Test that we can learn some lines and get a word back."""
|
||||
learn_line("the elephant goes ERRRRRRRRRRRRR", MarkovContext.objects.get(pk=1))
|
||||
word = get_word_out_of_states(MarkovState.objects.all())
|
||||
self.assertIn(word, ['the', 'elephant', 'goes', 'ERRRRRRRRRRRRR', '__stop'])
|
Loading…
Reference in New Issue
Block a user