DBDM_Project/UnitTest.py at master · EnguerrandPrebet/DBDM_Project · GitHub

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
45
46
47
48
49
50
51
import algo1,unittest

from classe import *

class MyTestCaseNaive(unittest.TestCase):
	def test_normal_case(self):
		fd = set([FD(SetAttr('A'),SetAttr('B')),FD(SetAttr('B'),SetAttr('C')),FD(SetAttr('C'),SetAttr('D')),FD(SetAttr('D'),SetAttr('E'))])
		attr=SetAttr('A')
		self.assertEqual(algo1.naive(fd, attr), SetAttr(['A','B','C','D','E']))
	def test_empty_fds(self):
		fd = set()
		attr=SetAttr('A')
		self.assertEqual(algo1.naive(fd, attr), SetAttr('A'))
	def test_right_blank(self):
		fd = set([FD(SetAttr('A'),SetAttr()),FD(SetAttr('B'),SetAttr('C'))])
		attr=SetAttr('A')
		self.assertEqual(algo1.naive(fd, attr), SetAttr('A'))
	def test_left_blank(self):
		fd = set([FD(SetAttr('A'),SetAttr('B')),FD(SetAttr('B'),SetAttr('C')),FD(SetAttr(),SetAttr('D'))])
		attr=SetAttr('A')
		self.assertEqual(algo1.naive(fd, attr), SetAttr(['A','B','C','D']))
	def test_compliquer_case(self):
		fd = set([FD(SetAttr('A'),SetAttr(['B','C'])),FD(SetAttr('C'),SetAttr(['D','E'])),FD(SetAttr(['B','E']),SetAttr('F')),FD(SetAttr('A'),SetAttr('C'))])
		attr=SetAttr('A')
		self.assertEqual(algo1.naive(fd, attr), SetAttr(['A','B','C','D','E','F']))


class MyTestCaseimproved(unittest.TestCase):
	def test_normal_case(self):
		fd = set([FD(SetAttr('A'),SetAttr('B')),FD(SetAttr('B'),SetAttr('C')),FD(SetAttr('C'),SetAttr('D')),FD(SetAttr('D'),SetAttr('E'))])
		attr=SetAttr('A')
		self.assertEqual(algo1.improved(fd, attr), SetAttr(['A','B','C','D','E']))
	def test_empty_fds(self):
		fd = set()
		attr=SetAttr('A')
		self.assertEqual(algo1.improved(fd, attr), SetAttr('A'))
	def test_right_blank(self):
		fd = set([FD(SetAttr('A'),SetAttr()),FD(SetAttr('B'),SetAttr('C'))])
		attr=SetAttr('A')
		self.assertEqual(algo1.improved(fd, attr), SetAttr('A'))
	def test_left_blank(self):
		fd = set([FD(SetAttr('A'),SetAttr('B')),FD(SetAttr('B'),SetAttr('C')),FD(SetAttr(),SetAttr('D'))])
		attr=SetAttr('A')
		self.assertEqual(algo1.improved(fd, attr), SetAttr(['A','B','C','D']))
	def test_compliquer_case(self):
		fd = set([FD(SetAttr('A'),SetAttr(['B','C'])),FD(SetAttr('C'),SetAttr(['D','E'])),FD(SetAttr(['B','E']),SetAttr('F')),FD(SetAttr('A'),SetAttr('C'))])
		attr=SetAttr('A')
		self.assertEqual(algo1.improved(fd, attr), SetAttr(['A','B','C','D','E','F']))

if __name__ == '__main__':
	unittest.main()