def test_simple(self):
f = ReadableFileMock(csv_file_data1)
csv_reader = RecordNominalCSVReader(f, header=True)
relation = Relation()
relation.read(csv_reader)
self.assertSetEqual(relation.attributes[2].values,
set(['Y', 'N']))
self.assertEqual(relation[1], ('2', '4', 'Y'))
self.assertEqual(len(relation), 5)
relation2 = relation[1:3]
self.assertIsInstance(relation2, Relation)
self.assertEqual(relation2[1], ('3', '4', 'N'))
self.assertEqual(len(relation.sample_without_replacement(2)), 2)
self.assertEqual(len(relation.sample_with_replacement(2)), 2)
split1 = relation.split_by_attribute_values(0)
self.assertEqual(len(split1), 5)
split2 = relation.split_by_attribute_values(1)
self.assertEqual(len(split2), 2)
self.assertEqual(len(split2['2']), 2)
self.assertEqual(len(split2['4']), 3)
split3 = relation.split_by_attribute_values(2)
self.assertEqual(len(split3), 2)
self.assertEqual(len(split3['Y']), 2)
self.assertEqual(len(split3['N']), 3)
self.assertListEqual(relation.attribute_values(0),
['1', '2', '3', '5', '6'])
self.assertListEqual(relation.attribute_values(2),
['Y', 'Y', 'N', 'N', 'N'])
