<filename>test/visualization/test_visualize.py<gh_stars>100-1000
# coding: utf-8

from pathlib import Path

import pytest

from AxonDeepSeg.visualization.visualize import visualize_training


class TestCore(object):
    def setup(self):
        # Get the directory where this current file is saved
        self.fullPath = Path(__file__).resolve().parent
        # Move up to the test directory, "test/"
        self.testPath = self.fullPath.parent

        self.pathModel = (
            self.testPath /
            '__test_files__' /
            '__test_model__' /
            'Model'
            )

    def teardown(self):
        pass

    # --------------visualize_training tests-------------- #
    @pytest.mark.unit
    def test_visualize_training_runs_successfully(self):

        assert visualize_training(str(self.pathModel))

<filename>test/visualization/test_visualize.py<gh_stars>100-1000
# coding: utf-8

from pathlib import Path

import pytest

from AxonDeepSeg.visualization.visualize import visualize_training


class TestCore(object):
    def setup(self):
        # Get the directory where this current file is saved
        self.fullPath = Path(__file__).resolve().parent
        # Move up to the test directory, "test/"
        self.testPath = self.fullPath.parent

        self.pathModel = (
            self.testPath /
            '__test_files__' /
            '__test_model__' /
            'Model'
            )

    def teardown(self):
        pass

    # --------------visualize_training tests-------------- #
    @pytest.mark.unit
    def test_visualize_training_runs_successfully(self):

        assert visualize_training(str(self.pathModel))

# coding: utf-8 # Get the directory where this current file is saved # Move up to the test directory, "test/" # --------------visualize_training tests-------------- #