emmental/curds/test_data.py

32 lines
884 B
Python

# Copyright 2019 (c) Anna Schumaker.
import data
import os
import unittest
import xdg.BaseDirectory
xdg_data_home = xdg.BaseDirectory.xdg_data_home
testing_data = os.path.join(xdg_data_home, "emmental-testing")
class Test:
def __init__(self):
self.one = 1
self.two = 2
self.list = [ 3, 4, 5]
class TestDataModule(unittest.TestCase):
def setUp(self):
path = os.path.join(testing_data, "test")
if os.path.exists(path):
os.remove(path)
def test_dir(self):
self.assertEqual(data.emmental_data, testing_data)
self.assertTrue(os.path.exists(testing_data))
self.assertTrue(os.path.isdir(testing_data))
def test_data_file_path(self):
path = os.path.join(testing_data, "test")
self.assertEqual(data.data_file_path("test"), path)
self.assertFalse(os.path.exists(path))