devops-exercises/tests/scripts_question_utils_unittest.py

32 lines
967 B
Python
Raw Normal View History

2019-12-24 19:34:31 +01:00
import unittest
from pathlib import Path
from typing import List
2020-01-09 12:12:06 +01:00
from scripts.question_utils import get_answered_questions, get_question_list
2019-12-24 19:34:31 +01:00
def open_test_case_file(n: int) -> List[bytes]:
tests_path = Path(__file__).parent.joinpath()
with open(f'{tests_path}/testcases/testcase{n}.md', 'rb') as f:
file_list = [line.rstrip() for line in f.readlines()]
return file_list
class QuestionCount(unittest.TestCase):
def test_case_1(self):
raw_list = open_test_case_file(1)
question_list = get_question_list(raw_list)
answers = get_answered_questions(question_list)
self.assertEqual(len(question_list), 11)
self.assertEqual(answers, 3)
def test_case_2(self):
raw_list = open_test_case_file(2)
question_list = get_question_list(raw_list)
answers = get_answered_questions(question_list)
self.assertEqual(len(question_list), 16)
self.assertEqual(answers, 11)