1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44
|
import unittest
import arff
class TestEncodeRelation(unittest.TestCase):
def get_encoder(self):
decoder = arff.ArffEncoder()
return decoder
def test_simple(self):
'''Relation name without spaces.'''
encoder = self.get_encoder()
fixture = u'relation-name'
result = encoder._encode_relation(fixture)
expected = u'@RELATION relation-name'
self.assertEqual(result, expected)
def test_espaced(self):
'''Relation name with spaces.'''
encoder = self.get_encoder()
fixture = u'relation name and'
result = encoder._encode_relation(fixture)
expected = u'@RELATION "relation name and"'
self.assertEqual(result, expected)
def test_special(self):
'''Relation name with spaces.'''
encoder = self.get_encoder()
fixture = u'%relationnameand'
result = encoder._encode_relation(fixture)
expected = u'@RELATION "%relationnameand"'
self.assertEqual(result, expected)
fixture = u'relation,nameand'
result = encoder._encode_relation(fixture)
expected = u'@RELATION "relation,nameand"'
self.assertEqual(result, expected)
|