from psychopy import core
from psychopy import misc
from psychopy.visual.windowwarp import Warper
from psychopy.monitors import Monitor
from psychopy.visual.grating import GratingStim
from psychopy.visual.window import Window
mon = Monitor('GenericMonitor', width=33.169, distance=10)
mon.setSizePix((1440, 900))
win = Window((1440, 900),
monitor=mon,
fullscr=True,
useFBO=True,
allowStencil=True)
warper = Warper(win, warp='spherical', warpGridsize=300, eyepoint=[0.5, 0.5])
stim = GratingStim(win=win,
units='deg',
tex='sin',
sf=0.1,
size=misc.pix2deg(win.size, win.monitor))
print win.size
print 'win size', win.size
print 'mon size', win.monitor.getSizePix()
print 'as deg', misc.pix2deg(win.size, win.monitor)
stim.draw()
win.flip()
core.wait(0.5)
win.close()
class Test_Form(object):
"""Test suite for Form component"""
def setup_class(self):
# Create temp files for storing items
self.temp_dir = mkdtemp()
self.fileName_xlsx = os.path.join(self.temp_dir, 'items.xlsx')
self.fileName_csv = os.path.join(self.temp_dir, 'items.csv')
# create some questions
self.questions = []
self.genderItem = {"questionText": "What is your gender?",
"questionWidth": 0.7,
"type": "radio",
"responseWidth": 0.3,
"options": "Male, Female, Other",
"layout": 'vert',
"index": 0,
"questionColor": "white",
"responseColor": "white"
}
self.questions.append(self.genderItem)
# then a set of ratings
items = ["running", "cake", "programming"]
for idx, item in enumerate(items):
entry = {"questionText": "How much you like {}".format(item),
"questionWidth": 0.7,
"type": "rating",
"responseWidth": 0.3,
"options":"Lots, some, Not a lot, Longest Option",
"layout": 'horiz',
"index": idx+1,
"questionColor": "white",
"responseColor": "white"
}
self.questions.append(entry)
self.win = Window(units='height', allowStencil=True, autoLog=False)
self.survey = Form(self.win, items=self.questions, size=(1.0, 0.3), pos=(0.0, 0.0), autoLog=False)
# Create datafiles
df = DataFrame(self.questions)
df.to_excel(self.fileName_xlsx, index=False)
df.to_csv(self.fileName_csv, index=False)
def test_importItems(self):
wrongFields = [{"a": "What is your gender?",
"b": 0.7,
"c": "radio",
"d": 0.3,
"e": "Male, Female, Other",
"f": 'vert',
"g": "white",
"h": "white"
}]
# this doesn't include a itemText or questionText so should get an err
missingHeader = [{"qText": "What is your gender?",
"questionWidth": 0.7,
"type": "radio",
"responseWidth": 0.3,
"options": "Other",
"layout": 'vert',
"index": 0,
"questionColor": "white",
"responseColor": "white"}]
# Check options for list of dicts
with pytest.raises(ValueError):
self.survey = Form(self.win, items=missingHeader, size=(1.0, 0.3), pos=(0.0, 0.0), autoLog=False)
# Check csv
self.survey = Form(self.win, items=self.fileName_csv,
size=(1.0, 0.3), pos=(0.0, 0.0), autoLog=False)
# Check Excel
self.survey = Form(self.win, items=self.fileName_xlsx,
size=(1.0, 0.3), pos=(0.0, 0.0), randomize=False, autoLog=False)
def test_set_scroll_speed(self):
items = 2
for multipliers in [1,2,3,4]:
assert self.survey.setScrollSpeed([0] * items, multipliers) == items * multipliers
assert self.survey.setScrollSpeed([0] * items, multipliers) == items * multipliers
assert self.survey.setScrollSpeed([0] * items, multipliers) == items * multipliers
def test_response_text_wrap(self):
options = ['a', 'b', 'c']
for size in [.2, .3, .4]:
item = {"responseWidth": size, "options": options}
def test_set_questions(self):
survey = Form(self.win, items=[self.genderItem], size=(1.0, 0.3),
pos=(0.0, 0.0), autoLog=False)
ctrl, h, w = survey._setQuestion(self.genderItem)
assert type(ctrl) == TextBox2
assert type(h) in [float, np.float64]
assert type(w) in [float, np.float64]
def test_set_response(self):
survey = Form(self.win, items=[self.genderItem], size=(1.0, 0.3),
pos=(0.0, 0.0), autoLog=False)
ctrl, h, w = survey._setQuestion(self.genderItem)
sliderStim, respHeight = survey._setResponse(self.genderItem)
assert type(sliderStim) == Slider
assert type(respHeight) in [float, np.float64]
def test_form_size(self):
assert self.survey.size[0] == (1.0, 0.3)[0] # width
assert self.survey.size[1] == (1.0, 0.3)[1] # height
def test_aperture_size(self):
assert self.survey.aperture.size[0] == self.survey.size[0]
assert self.survey.aperture.size[1] == self.survey.size[1]
def test_border_limits(self):
survey = self.survey
assert survey.leftEdge == survey.pos[0] - survey.size[0]/2.0
assert survey.rightEdge == survey.pos[0] + survey.size[0]/2.0
assert survey.topEdge == survey.pos[1] + survey.size[1]/2.0
def test_text_height(self):
assert self.survey.textHeight == 0.02
def test_font(self):
exemplars = [
{"file": "form_font_demographics.xlsx", "font": "Open Sans",
"screenshot": "form_font_demographics.png"},
]
tykes = [
{"file": "form_font_demographics.xlsx", "font": "Indie Flower",
"screenshot": "form_font_nonstandard.png"},
{"file": "form_font_languages.xlsx", "font": "Rubrik",
"screenshot": "form_font_languages.png"},
]
for case in exemplars + tykes:
survey = Form(self.win, items=str(Path(utils.TESTS_DATA_PATH) / case['file']), size=(1, 1), font=case['font'],
fillColor=None, borderColor=None, itemColor="white", responseColor="white", markerColor="red",
pos=(0, 0), autoLog=False)
survey.draw()
self.win.flip()
# self.win.getMovieFrame(buffer='front').save(Path(utils.TESTS_DATA_PATH) / case['screenshot'])
utils.compareScreenshot(Path(utils.TESTS_DATA_PATH) / case['screenshot'], self.win, crit=20)
def test_item_padding(self):
assert self.survey.itemPadding == 0.05
def test_form_units(self):
assert self.survey.units == 'height'
def test_screen_status(self):
"""Test whether the object is visible"""
assert self.survey._inRange(self.survey.items[0]['itemCtrl'])
def test_get_data(self):
self.survey = Form(self.win, items=self.questions, size=(1.0, 0.3),
pos=(0.0, 0.0), autoLog=False)
data = self.survey.getData()
Qs = [this['itemText'] for this in data]
indices = [item['index'] for item in data]
assert Qs == ['What is your gender?',
'How much you like running',
'How much you like cake',
'How much you like programming',]
assert all([item['response'] is None for item in data])
assert all([item['rt'] is None for item in data])
assert list(indices) == list(range(4))
def teardown_class(self):
shutil.rmtree(self.temp_dir)
self.win.close()
def test_open_glfw_window():
from psychopy.visual.window import Window
win = Window(winType='glfw', autoLog=False)
assert win.winType == 'glfw'
win.flip()
win.close()
class Test_Slider(_TestColorMixin):
def setup_class(self):
self.win = Window([128, 128],
pos=[50, 50],
monitor="testMonitor",
allowGUI=False,
autoLog=False)
self.obj = Slider(self.win,
units="height",
size=(1, 0.1),
pos=(0, 0.5),
style='radio')
self.obj.markerPos = 1
# Pixel which is the border color
self.borderPoint = (0, 127)
self.borderUsed = True
# Pixel which is the fill color
self.fillPoint = (0, 0)
self.fillUsed = True
# Pixel which is the fore color
self.forePoint = (0, 0)
self.foreUsed = False
def teardown_class(self):
self.win.close()
def test_horiz(self):
# Define cases
exemplars = [
{
'size': (1, 0.2),
'ori': 0,
'horiz': True,
'tag': 'horiz'
}, # Wide slider, no rotation
{
'size': (0.2, 1),
'ori': 0,
'horiz': False,
'tag': 'vert'
}, # Tall slider, no rotation
{
'size': (1, 0.2),
'ori': 90,
'horiz': False,
'tag': 'vert'
}, # Wide slider, 90deg rotation
{
'size': (0.2, 1),
'ori': 90,
'horiz': True,
'tag': 'horiz'
}, # Tall slider, 90deg rotation
]
tykes = [
{
'size': (1, 0.2),
'ori': 25,
'horiz': True,
'tag': 'accute_horiz'
}, # Wide slider, accute rotation
{
'size': (0.2, 1),
'ori': 25,
'horiz': False,
'tag': 'accute_vert'
}, # Tall slider, accute rotation
{
'size': (1, 0.2),
'ori': 115,
'horiz': False,
'tag': 'obtuse_horiz'
}, # Wide slider, obtuse rotation
{
'size': (0.2, 1),
'ori': 115,
'horiz': True,
'tag': 'obtuse_vert'
}, # Tall slider, obtuse rotation
]
# Try each case
self.win.flip()
for case in exemplars + tykes:
# Make sure horiz is set as intended
obj = Slider(self.win,
labels=["a", "b", "c", "d"],
ticks=[1, 2, 3, 4],
labelHeight=0.2,
labelColor='red',
size=case['size'],
ori=case['ori'])
assert obj.horiz == case['horiz']
# Make sure slider looks as intended
obj.draw()
filename = f"test_slider_horiz_{case['tag']}.png"
# self.win.getMovieFrame(buffer='back').save(Path(utils.TESTS_DATA_PATH) / filename)
utils.compareScreenshot(Path(utils.TESTS_DATA_PATH) / filename,
self.win,
crit=10)
self.win.flip()
def test_reset(self):
s = Slider(self.win, size=(1, 0.1))
s.markerPos = 1
s.history = [1]
s.rating = 1
s.rt = 1
s.status = constants.STARTED
s.reset()
assert s.markerPos == None
assert s.history == []
assert s.rating == None
assert s.rt == None
assert s.status == constants.NOT_STARTED
def test_elements(self):
s = Slider(self.win, size=(1, 0.1))
assert type(s.line) == type(Rect(self.win))
assert type(s.tickLines) == type(ElementArrayStim(self.win))
assert type(s.marker) == type(ShapeStim(self.win))
assert type(s.validArea) == type(Rect(self.win))
def test_pos(self):
s = Slider(self.win, size=(1, 0.1))
positions = [(.05, .05), (0.2, .2)]
for newPos in positions:
s.pos = newPos
assert array_equal(s.pos, newPos)
def test_ratingToPos(self):
s = Slider(
self.win,
size=(1, 0.1),
)
assert s._ratingToPos(3)[0] == 0
assert s._ratingToPos(1)[0] == -.5
assert s._ratingToPos(5)[0] == .5
def test_posToRatingToPos(self):
s = Slider(
self.win,
size=(1, 0.1),
)
assert s._posToRating((0, 0)) == 3
assert s._posToRating((-.5, 0)) == 1
assert s._posToRating((.5, 0)) == 5
def test_tick_and_label_locs(self):
exemplars = [
{
'ticks': [1, 2, 3, 4, 5],
'labels': ["a", "b", "c", "d", "e"],
'tag': "simple"
},
{
'ticks': [1, 2, 3, 9, 10],
'labels': ["a", "b", "c", "d", "e"],
'tag': "clustered"
},
{
'ticks': [1, 2, 3, 4, 5],
'labels': ["", "b", "c", "d", ""],
'tag': "blanks"
},
{
'ticks': None,
'labels': ["a", "b", "c", "d", "e"],
'tag': "noticks"
},
{
'ticks': [1, 2, 3, 4, 5],
'labels': None,
'tag': "nolabels"
},
]
tykes = [
{
'ticks': [1, 2, 3],
'labels': ["a", "b", "c", "d", "e"],
'tag': "morelabels"
},
{
'ticks': [1, 2, 3, 4, 5],
'labels': ["a", "b", "c"],
'tag': "moreticks"
},
{
'ticks': [1, 9, 10],
'labels': ["a", "b", "c", "d", "e"],
'tag': "morelabelsclustered"
},
{
'ticks': [1, 7, 8, 9, 10],
'labels': ["a", "b", "c", "d"],
'tag': "moreticksclustered"
},
]
# Test all cases
self.win.flip()
for case in exemplars + tykes:
# Make vertical slider
vert = Slider(self.win,
size=(0.1, 0.5),
pos=(-0.25, 0),
units="height",
labels=case['labels'],
ticks=case['ticks'])
vert.draw()
# Make horizontal slider
horiz = Slider(self.win,
size=(0.5, 0.1),
pos=(0.2, 0),
units="height",
labels=case['labels'],
ticks=case['ticks'])
horiz.draw()
# Compare screenshot
filename = "test_slider_ticklabelloc_%(tag)s.png" % case
#self.win.getMovieFrame(buffer='back').save(Path(utils.TESTS_DATA_PATH) / filename)
utils.compareScreenshot(
Path(utils.TESTS_DATA_PATH) / filename, self.win)
self.win.flip()
def test_granularity(self):
s = Slider(self.win, size=(1, 0.1), granularity=1)
minRating, maxRating = 1, 5
assert s.granularity == 1
assert s._granularRating(1) == minRating
assert s._granularRating(5) == maxRating
assert s._granularRating(0) == minRating
assert s._granularRating(6) == maxRating
def test_rating(self):
s = Slider(self.win, size=(1, 0.1))
minRating, maxRating = 1, 5
s.rating = 1
assert s.rating == minRating
s.rating = 5
assert s.rating == maxRating
s.rating = 0
assert s.rating == minRating
s.rating = 6
assert s.rating == maxRating
def test_markerPos(self):
s = Slider(self.win, size=(1, 0.1))
s._updateMarkerPos = False
minPos, maxPos = 1, 5
assert s._updateMarkerPos != True
s.markerPos = 1
assert s.markerPos == minPos
s.markerPos = 5
assert s.markerPos == maxPos
s.markerPos = 0
assert s.markerPos == minPos
s.markerPos = 6
assert s.markerPos == maxPos
assert s._updateMarkerPos == True
def test_recordRating(self):
s = Slider(self.win, size=(1, 0.1))
minRating, maxRating = 1, 5
counter = 0
for rates in range(0, 7):
s.recordRating(rates, random.random())
counter += 1
ratings = [rating[0] for rating in s.history]
RT = [rt[1] for rt in s.history]
assert len(s.history) == counter
assert len(ratings) == counter
assert min(ratings) == minRating
assert max(ratings) == maxRating
assert len(RT) == counter
assert max(RT) <= 1
assert min(RT) >= 0
def test_getRating(self):
s = Slider(self.win, size=(1, 0.1))
minRating, maxRating = 1, 5
s.rating = 1
assert s.getRating() == minRating
s.rating = 5
assert s.getRating() == maxRating
s.rating = 0
assert s.getRating() == minRating
s.rating = 6
assert s.getRating() == maxRating
def test_getRT(self):
s = Slider(self.win, size=(1, 0.1))
testRT = .89
s.recordRating(2, testRT)
assert s.history[-1][1] == s.getRT()
assert type(s.getRT()) == float
assert s.getRT() == testRT
def test_open_glfw_window():
win = Window(winType='glfw', autoLog=False)
assert win.winType == 'glfw'
win.flip()
win.close()