Add new test for improving coverage

- Install pytest on github actions.
- Replace unittest for pytest on github actions.

.github/workflows/main.yml

@@ -21,10 +21,12 @@ jobs:
  with:
  python-version: '3.6'
  - name: Install dependencies
- run: pip install coverage
+ run: | 
+ pip install coverage
+ pip install -U pytest
  - name: Test unittest
  run: |
- coverage run -m unittest discover
+ coverage run -m pytest
  coverage xml
  - name: "Upload coverage to Codecov"
  uses: codecov/codecov-action@v2

.vscode/settings.json

@@ -0,0 +1,7 @@
+{
+ "python.testing.pytestArgs": [
+ "tests"
+ ],
+ "python.testing.unittestEnabled": false,
+ "python.testing.pytestEnabled": true
+}

mdutils/fileutils/fileutils.py

@@ -22,13 +22,16 @@ def __init__(self, name='', dirname: str = None):
  If not you can specify the path directly on the name."""
  if name:
  self.dirname = dirname
- if dirname:
- self.file_name = f'{dirname}/{name}' if name.endswith('.md') else f'{dirname}/{name}.md'
- else:
- self.file_name = name if name.endswith('.md') else f'{name}.md'
+ self.file_name = self._get_file_name(name, dirname)
  self.file = open(f'{self.file_name}', 'w+', encoding='UTF-8')
  self.file.close()
 
+ def _get_file_name(self, name: str, dirname: str = None ) -> str:
+ if dirname:
+ return f'{dirname}/{name}' if name.endswith('.md') else f'{dirname}/{name}.md'
+
+ return name if name.endswith('.md') else f'{name}.md'
+
  def rewrite_all_file(self, data):
  """Rewrite all the data of a Markdown file by ``data``.
 
@@ -57,7 +60,7 @@ def append_after_second_line(self, data):
  self.file.write('\n' + file_data[len(first_line + second_line):])
 
  @staticmethod
- def read_file(file_name):
+ def read_file(file_name: str) -> str:
  """Read a Markdown file using a file name. It is not necessary to add *.md extension.
 
  :param file_name: Markdown file's name.

tests/test_fileutils/test_fileutils.py

@@ -9,25 +9,29 @@
 from unittest import TestCase
 from mdutils.fileutils import MarkDownFile
 import tempfile
-import os
 
 
 class TestMarkdownFile(TestCase):
 
  def test_create_file(self):
  with tempfile.TemporaryDirectory() as tmp:
  file = MarkDownFile('Test_file', tmp)
- self.assertEqual(file.file_name, 'Test_file.md')
+ self.assertEqual(file.file_name, f'{tmp}/Test_file.md')
 
  def test_create_file_case_0(self):
  with tempfile.TemporaryDirectory() as tmp:
  file = MarkDownFile('Test_filemd', tmp)
- self.assertEqual(file.file_name, 'Test_filemd.md')
+ self.assertEqual(file.file_name, f'{tmp}/Test_filemd.md')
 
  def test_create_file_case_1(self):
  with tempfile.TemporaryDirectory() as tmp:
  file = MarkDownFile('Test_file.md', tmp)
- self.assertEqual(file.file_name, 'Test_file.md')
+ self.assertEqual(file.file_name, f'{tmp}/Test_file.md')
+
+ def test_create_file_case_3(self):
+ with tempfile.TemporaryDirectory() as tmp:
+ file = MarkDownFile(f'{tmp}/Test_file.md')
+ self.assertEqual(file.file_name, f'{tmp}/Test_file.md')
 
  def test_rewrite_all_file(self):
  expected_content = "Write some content"