Total Complexity | 5 |
Total Lines | 69 |
Duplicated Lines | 0 % |
Changes | 0 |
1 | import os |
||
2 | from sources.db_extractor.BasicNeeds import BasicNeeds |
||
3 | # package to facilitate multiple operation system operations |
||
4 | import platform |
||
5 | import unittest |
||
6 | |||
7 | |||
8 | class MyTestCase(unittest.TestCase): |
||
9 | |||
10 | def test_file_statistics(self): |
||
11 | parent_folder = os.path.abspath(os.pardir) + '/' + '/'.join([ |
||
12 | 'virtual_environment', |
||
13 | 'Scripts' |
||
14 | ]) |
||
15 | relevant_file = 'pip' |
||
16 | if platform.system() == 'Windows': |
||
17 | relevant_file += '.exe' |
||
18 | file_to_evaluate = os.path.join(parent_folder, relevant_file) |
||
19 | bn = BasicNeeds() |
||
20 | value_to_assert = bn.fn_get_file_statistics(file_to_evaluate)['size [bytes]'] |
||
21 | value_to_compare_with = os.path.getsize(file_to_evaluate) |
||
22 | self.assertEqual(value_to_assert, value_to_compare_with) |
||
23 | |||
24 | def test_numbers_with_leading_zero(self): |
||
25 | pair_values = [ |
||
26 | { |
||
27 | 'given': '1', |
||
28 | 'leading_zeros': 2, |
||
29 | 'expected': '01' |
||
30 | }, |
||
31 | { |
||
32 | 'given': '1', |
||
33 | 'leading_zeros': 10, |
||
34 | 'expected': '0000000001' |
||
35 | }, |
||
36 | { |
||
37 | 'given': '1', |
||
38 | 'leading_zeros': 1, |
||
39 | 'expected': '1' |
||
40 | } |
||
41 | ] |
||
42 | bn = BasicNeeds() |
||
43 | for current_pair in pair_values: |
||
44 | |||
45 | value_to_assert = bn.fn_numbers_with_leading_zero(current_pair['given'], |
||
46 | current_pair['leading_zeros']) |
||
47 | self.assertEqual(value_to_assert, current_pair['expected'], |
||
48 | 'Provided values was "' + value_to_assert + '", Expected') |
||
49 | pair_failing_values = [ |
||
50 | { |
||
51 | 'given': '1', |
||
52 | 'leading_zeros': 2, |
||
53 | 'expected': '02' |
||
54 | }, |
||
55 | { |
||
56 | 'given': '1', |
||
57 | 'leading_zeros': 0, |
||
58 | 'expected': '' |
||
59 | } |
||
60 | ] |
||
61 | for current_pair in pair_failing_values: |
||
62 | value_to_assert = bn.fn_numbers_with_leading_zero(current_pair['given'], |
||
63 | current_pair['leading_zeros']) |
||
64 | self.assertNotEqual(value_to_assert, current_pair['expected']) |
||
65 | |||
66 | |||
67 | if __name__ == '__main__': |
||
68 | unittest.main() |
||
69 |