Best Python code snippet using grail_python
sign_up.py
Source:sign_up.py  
...23    error_id = set_error_message(message=message)24    return redirect(url_for("auth.error", error=error_id))25@bp.get("/step1")26@login_block27@verify_step(1)28def step1():29    email = session['social.kakao.email']30    code_id = session['social.kakao.code_id']31    code: Code = Code.query.filter_by(32        id=code_id,33        email=email34    ).first()35    if code is None:36        return to(message="ë§ë£ë ì´ë©ì¼ ì¸ì¦ ìì²ì
ëë¤. ë¤ì ìëí´ì£¼ì¸ì.")37    if code.is_expired() or code.is_used():38        return to(message="ë§ë£ë ì´ë©ì¼ ì¸ì¦ ìì²ì
ëë¤. ë¤ì ìëí´ì£¼ì¸ì.")39    return render_template(40        "kakao/sign-up/step1.html",41        error=get_error_message(),42        email=environ['SMTP_USER']43    )44@bp.post("/step1")45@verify_step(1)46def step1_post():47    email = session['social.kakao.email']48    code_id = session['social.kakao.code_id']49    code: Code = Code.query.filter_by(50        id=code_id,51        email=email52    ).first()53    if code is None:54        return to(message="ë§ë£ë ì´ë©ì¼ ì¸ì¦ ìì²ì
ëë¤. ë¤ì ìëí´ì£¼ì¸ì.")55    if code.is_expired() or code.is_used():56        return to(message="ë§ë£ë ì´ë©ì¼ ì¸ì¦ ìì²ì
ëë¤. ë¤ì ìëí´ì£¼ì¸ì.")57    if code.code == request.form.get("code", "undefined"):58        session['social.kakao.step'] = 259        # used in step160        del session['social.kakao.code_id']61        return redirect(url_for("social.kakao.sign_up.step2"))62    error_id = set_error_message(message="ì¸ì¦ ì½ëê° ì¬ë°ë¥´ì§ ììµëë¤.")63    return redirect(url_for("social.kakao.sign_up.step1", error=error_id))64@bp.get("/step2")65@login_block66@verify_step(2)67def step2():68    try:69        email = session['social.kakao.email']70    except KeyError:71        return to(message="ì¬ë°ë¥¸ ìì²ì´ ìëëë¤")72    tos = TermsOfService.query.order_by(73        TermsOfService.id.desc()74    ).with_entities(75        TermsOfService.id76    ).first()77    privacy = PrivacyPolicy.query.order_by(78        PrivacyPolicy.id.desc()79    ).with_entities(80        PrivacyPolicy.id81    ).first()82    if tos is None or privacy is None:83        return "ìë¹ì¤ ì´ì©ì½ê´ê³¼ ê°ì¸ì ë³´ ì²ë¦¬ë°©ì¹¨ì´ ë±ë¡ëì§ ìì íìê°ì
ì ì§í í  ì ììµëë¤."84    session['social.kakao.version'] = {85        "tos": tos.id,86        "privacy": privacy.id87    }88    return render_template(89        "kakao/sign-up/step2.html",90        error=get_error_message()91    )92@bp.post("/step2")93@login_block94@verify_step(2)95def step2_post():96    def _to(message: str):97        error_id = set_error_message(message=message)98        return redirect(url_for("social.kakao.sign_up.step2", error=error_id))99    tos_agree = request.form.get("tos_agree", None) == "on"100    privacy_agree = request.form.get("privacy_agree", None) == "on"101    if tos_agree is False:102        return _to(message="ìë¹ì¤ ì´ì©ì½ê´ì ëìí´ì¼ ìë¹ì¤ë¥¼ ì´ì© í  ì ììµëë¤.")103    if privacy_agree is False:104        return _to(message="ê°ì¸ì ë³´ ì²ë¦¬ë°©ì¹¨ì ëìí´ì¼ ìë¹ì¤ë¥¼ ì´ì© í  ì ììµëë¤.")105    try:106        id = session['social.kakao.id']107        email = session['social.kakao.email']108        version = session['social.kakao.version']...test_nn.py
Source:test_nn.py  
...18            out = self.conv(x)19            return out20    shape = [1, 1, 28, 28]21    x = torch.randn(*shape)22    verify_step(Model(), [x])23def test_linear():24    class Model(nn.Module):25        def __init__(self):26            super().__init__()27            self.linear = nn.Linear(120, 84)28        def forward(self, x):29            out = self.linear(x)30            return out31    shape = [32, 120]32    x = torch.randn(*shape)33    verify_step(Model(), [x])34def test_sum():35    class Model(nn.Module):36        def __init__(self):37            super().__init__()38        def forward(self, x):39            out = torch.sum(x)40            return out41    shape = [32, 120]42    x = torch.randn(*shape)43    verify_step(Model(), [x], jit_script=False, tol=5e-4)44def test_pad():45    class Model(nn.Module):46        def __init__(self):47            super().__init__()48        def forward(self, x):49            pad = (1, 2, 3, 4, 5, 6)50            out = torch.nn.functional.pad(x, pad, "constant", 2)51            return out52    shape = [32, 120, 20]53    x = torch.randn(*shape)54    verify_step(Model(), [x], jit_script=False)55def test_gelu():56    """GeLU supports approximation since https://github.com/pytorch/pytorch/pull/72826"""57    class Model(nn.Module):58        def __init__(self):59            super().__init__()60            self.gelu = torch.nn.GELU("none")61        def forward(self, x):62            return self.gelu(x)63    shape = [5, 5]64    t_x_cpu = torch.randn(shape, requires_grad=True)65    verify_step(Model(), [t_x_cpu], jit_script=False)66    verify_step(Model(), [t_x_cpu], jit_script=False, with_backward=True)67@pytest.mark.parametrize("dtype", [torch.float32, torch.float16])68@pytest.mark.parametrize("norm_type", [1, 2])69def test_embedding(dtype, norm_type):70    class Model(nn.Module):71        def __init__(self):72            super().__init__()73            self.embedding = nn.Embedding(10, 3, norm_type=norm_type, dtype=dtype)74        def forward(self, x_input):75            return self.embedding(x_input)76    x = torch.randint(10, (3, 3))77    verify_step(Model(), [x], jit_script=False)78def test_softmax():79    class Model(nn.Module):80        def __init__(self):81            super().__init__()82            self.softmax = nn.Softmax()83        def forward(self, x_input):84            return self.softmax(x_input)85    shape = (3, 3)86    t_x_cpu = torch.randn(shape, requires_grad=True)87    verify_step(Model(), [t_x_cpu], jit_script=False)88    verify_step(Model(), [t_x_cpu], jit_script=False, with_backward=True)89@pytest.mark.parametrize("dtype", [torch.float32])90@pytest.mark.parametrize("p", [0.2, 0.6])91def test_dropout(p, dtype):92    class Model(nn.Module):93        def __init__(self):94            super().__init__()95            self.dropout = nn.Dropout(p=p)96        def forward(self, x_input):97            return self.dropout(x_input)98    def check_dropout(x, y, dx=None, dy=None):99        x, y = x.cpu().detach().numpy(), y.cpu().detach().numpy()100        mask = y != 0101        expected = mask * x / (1 - p)102        check(expected, y)...MapMetricStepTests.py
Source:MapMetricStepTests.py  
1import unittest2from dto.CacheMetricsDTO import CacheMetricsDTO3from etl.Pipeline import Pipeline4from etl.PipelineStep import PipelineStep5from steps.MapMetricStep import MapMetricStep6class MockPipelineSetupStep(PipelineStep):7    def __init__(self):8        super().__init__(self.__class__.__name__)9    def run_step(self, pipeline_context):10        mock_payload = {}11        mock_payload['mock_key'] = 'mock_value'12        mock_dto = CacheMetricsDTO(mock_payload)13        metrics = [mock_dto]14        pipeline_context.add_var('metrics_list', metrics)15class MockVerifyPipelineStep(PipelineStep):16    def __init__(self, verify_key):17        super().__init__(self.__class__.__name__)18        self.__verify_key = verify_key19        self.__assert_flag = False20    def run_step(self, pipeline_context):21        self.__assert_flag = (pipeline_context.get_var('metrics_list')[0].get_metric_var(self.__verify_key) == 'mock_value')22    def get_assert_flag(self):23        return self.__assert_flag24class MapMetricsStepTests(unittest.TestCase):25    def test_step_maps_direct(self):26        pipeline = Pipeline()27        pipeline.add_step(MockPipelineSetupStep())28        pipeline.add_step(MapMetricStep('mock_key', 'mock_key'))29        verify_step = MockVerifyPipelineStep('mock_key')30        pipeline.add_step(verify_step)31        pipeline.run()32        self.assertEqual(verify_step.get_assert_flag(), True)33    def test_step_maps_rename(self):34        pipeline = Pipeline()35        pipeline.add_step(MockPipelineSetupStep())36        pipeline.add_step(MapMetricStep('mock_key', 'mock_key2'))37        verify_step = MockVerifyPipelineStep('mock_key2')38        pipeline.add_step(verify_step)39        pipeline.run()40        self.assertEqual(verify_step.get_assert_flag(), True)41if __name__ == '__main__':...Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.
You could also refer to video tutorials over LambdaTest YouTube channel to get step by step demonstration from industry experts.
Get 100 minutes of automation test minutes FREE!!
