33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
|
import os
|
||
|
|
from unittest import TestCase
|
||
|
|
|
||
|
|
from .. import parse_filename
|
||
|
|
|
||
|
|
|
||
|
|
class MiscTests(TestCase):
|
||
|
|
|
||
|
|
def check_filename(self, path, exp_fname, exp_bname, exp_format):
|
||
|
|
fname, bname, format = parse_filename(path)
|
||
|
|
self.assertEqual(fname, exp_fname)
|
||
|
|
self.assertEqual(bname, exp_bname)
|
||
|
|
self.assertEqual(format, exp_format)
|
||
|
|
|
||
|
|
def test_parse_filename(self):
|
||
|
|
|
||
|
|
# check format detection
|
||
|
|
self.check_filename("test.ipynb", "test.ipynb", "test", "json")
|
||
|
|
self.check_filename("test.json", "test.json", "test", "json")
|
||
|
|
self.check_filename("test.py", "test.py", "test", "py")
|
||
|
|
|
||
|
|
# check parsing an unknown format
|
||
|
|
self.check_filename("test.nb", "test.nb.ipynb", "test.nb", "json")
|
||
|
|
|
||
|
|
# check parsing a full file path
|
||
|
|
abs_path = os.path.abspath("test.ipynb")
|
||
|
|
basename, ext = os.path.splitext(abs_path)
|
||
|
|
self.check_filename(abs_path, abs_path, basename, "json")
|
||
|
|
|
||
|
|
# check parsing a file name containing dots
|
||
|
|
self.check_filename("test.nb.ipynb", "test.nb.ipynb", "test.nb",
|
||
|
|
"json")
|