Skip to content

Commit

Permalink
Modified test cases
Browse files Browse the repository at this point in the history
  • Loading branch information
VoodooCode14 committed May 6, 2022
1 parent f7c3113 commit b34284e
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 4 deletions.
11 changes: 8 additions & 3 deletions finn/tests/integration_tests/visualization/topoplot.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,10 +7,13 @@
import unittest
import numpy as np

import matplotlib
import matplotlib.pyplot as plt

import finn.visualization.topoplot as tp

class test_sfc(unittest.TestCase):
def test_sfc(self):
class test_topoplot(unittest.TestCase):
def test_topoplot(self):
##Generate some demo data
#Values to be plotted
values_1 = [ 0,
Expand Down Expand Up @@ -69,7 +72,7 @@ def test_sfc(self):
v_min = -3
v_max = 3

file_path = "./test.svg"
file_path = None
screen_channels = False
annotate_ch_names = False

Expand All @@ -86,6 +89,8 @@ def test_sfc(self):
#Plot with significances after MCC only
topo.run(np.asarray([values_1, values_2, values_3]).transpose(), ch_name_list, omit_channels, substitute_channels, v_min, v_max, v_border_values, v_border_labels, file_path, screen_channels, annotate_ch_names)

#plt.show(block = True)

assert(True) # if visualization routine ran until here, consider it successfull

if __name__ == '__main__':
Expand Down
2 changes: 1 addition & 1 deletion finn/visualization/topoplot.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
import skimage.filters

import matplotlib
matplotlib.use("Qt5Agg")
#matplotlib.use("Qt5Agg")
import matplotlib.markers
import matplotlib.pyplot as plt

Expand Down

0 comments on commit b34284e

Please sign in to comment.