emmental/trackdb/test_tags.py

# Copyright 2021 (c) Anna Schumaker.
from lib import tagstore
from . import tags
import unittest

class TestTags(unittest.TestCase):
    def test_tags_init(self):
        self.assertIsInstance(tags.Artist,  tagstore.TagStore)
        self.assertIsInstance(tags.Album,   tagstore.TagSuperStore)

    def test_tags_reset(self):
        tags.Artist.store  = {"a" : 1 }
        tags.Album.store   = {("a", "b") : 2 }

        tags.reset()
        self.assertEqual(tags.Artist.store,  { })
        self.assertEqual(tags.Album.store,   { })