devops-exercises/tests/scripts_question_utils_unittest.py
2020-01-09 12:13:31 +01:00

32 lines
967 B
Python

import unittest
from pathlib import Path
from typing import List
from scripts.question_utils import get_answered_questions, get_question_list
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)