File tree Expand file tree Collapse file tree 2 files changed +13
-2
lines changed
Expand file tree Collapse file tree 2 files changed +13
-2
lines changed Original file line number Diff line number Diff line change @@ -628,15 +628,22 @@ def parse_diff_given(self, node):
628628 PROPERTY = 0
629629 VALUES = 1
630630
631+ if sys .version_info >= (3 , 8 , 0 ):
632+ attr_name = "value"
633+ else :
634+ attr_name = "s"
635+
636+
631637 if len (node .args ) == 2 :
632638 if self .cur_inline_test .parameterized :
633639 raise MalformedException ("inline test: Parameterized inline tests currently do not support differential tests." )
634640 else :
635641 devices = []
636642 for elt in node .args [VALUES ].elts :
637- if elt .value not in {"cpu" , "cuda" , "mps" }:
643+ value = getattr (elt , attr_name )
644+ if value not in {"cpu" , "cuda" , "mps" }:
638645 raise MalformedException (f"Invalid device: { elt .value } . Must be one of ['cpu', 'cuda', 'mps']" )
639- devices .append (elt . value )
646+ devices .append (value )
640647 setattr (self .cur_inline_test , node .args [PROPERTY ].id , devices )
641648 else :
642649 raise MalformedException ("inline test: invalid diff_given(), expected 2 args" )
Original file line number Diff line number Diff line change 22from _pytest .pytester import Pytester
33import pytest
44
5+ # For testing in Spyder only
6+ if __name__ == "__main__" :
7+ pytest .main (['-v' , '-s' ])
8+
59# pytest -p pytester
610class TestInlinetests :
711 def test_inline_diff_given (self , pytester : Pytester ):
You can’t perform that action at this time.
0 commit comments