Best Python code snippet using green
tests.py
Source:tests.py
1"""2test.py3This file tests the various necessary util functions.4"""5from .io import *6import contextlib7import pandas as pd8import unittest9@contextlib.contextmanager10def set_env(environ):11 """12 Temporarily set the process environment variables. Taken from https://stackoverflow.com/questions/2059482/python-temporarily-modify-the-current-processs-environment/3433371013 >>> with set_env(PLUGINS_DIR=u'test/plugins'):14 ... "PLUGINS_DIR" in os.environ15 True16 >>> "PLUGINS_DIR" in os.environ17 False18 :type environ: dict[str, unicode]19 :param environ: Environment variables to set20 """21 old_environ = dict(os.environ)22 os.environ.update(environ)23 print(os.environ)24 try:25 yield26 finally:27 os.environ.clear()28 os.environ.update(old_environ)29class IOTests(unittest.TestCase):30 def setUp(self):31 self.toy_df = pd.DataFrame({'col1': [1, 2, 3], 'col2': [2, 4, 6]})32 self.fake_creds = {33 'AWS_ACCESS_KEY_ID': 'nothing',34 'AWS_SECRET_ACCESS_KEY': 'nothing',35 'AWS_DEFAULT_REGION': 'nothing',36 }37 def test_write_file_without_creds(self):38 pass39 def test_write_file_without_name(self):40 with self.assertRaises(AssertionError):41 write_file(self.toy_df, '')42 def test_write_file_scratch(self):43 filename = write_file(self.toy_df, 'test.pq')44 self.assertEqual(45 filename, 's3://toy-applied-ml-pipeline/scratch/test.pq')46 def test_write_file_no_scratch(self):47 filename = write_file(self.toy_df, 'test/test.pq', scratch=False)48 self.assertEqual(49 filename, 's3://toy-applied-ml-pipeline/test/test.pq')50 def test_save_output_without_creds(self):51 pass52 def test_save_output_no_version(self):53 filename = save_output_df(self.toy_df, 'test')54 self.assertIn('s3://toy-applied-ml-pipeline/dev/test/', filename)55 def test_save_output_no_component(self):56 with self.assertRaises(AssertionError):57 save_output_df(self.toy_df, '')58 def test_save_output_with_version(self):59 filename = save_output_df(self.toy_df, 'test',60 version='test', overwrite=True)61 self.assertEqual(62 filename, 's3://toy-applied-ml-pipeline/dev/test/test.pq')63 def test_save_output_no_overwrite(self):64 save_output_df(self.toy_df, 'test',65 version='test_no_overwrite', overwrite=True)66 with self.assertRaises(OSError):67 save_output_df(self.toy_df, 'test', version='test_no_overwrite')68 def test_save_output_overwrite(self):69 filename_1 = save_output_df(self.toy_df, 'test',70 version='test_overwrite', overwrite=True)71 filename_2 = save_output_df(self.toy_df, 'test',72 version='test_overwrite', overwrite=True)73 self.assertEqual(filename_1, filename_2)74if __name__ == '__main__':...
test_numeric.py
Source:test_numeric.py
...47 z = correlate(self.x, self.y, "full")48 assert_array_almost_equal(z, self.z1)49 z = correlate(self.y, self.x, "full")50 assert_array_almost_equal(z, self.z2)51 def test_no_overwrite(self):52 d = cp.ones(100)53 k = cp.ones(3)54 correlate(d, k)55 assert_array_equal(d, cp.ones(100))56 assert_array_equal(k, cp.ones(3))57 def test_complex(self):58 x = cp.array([1, 2, 3, 4 + 1j], dtype=complex)59 y = cp.array([-1, -2j, 3 + 1j], dtype=complex)60 r_z = cp.array(61 [3 - 1j, 6, 8 + 1j, 11 + 5j, -5 + 8j, -4 - 1j], dtype=complex62 )63 r_z = r_z[::-1].conj()64 z = correlate(y, x, mode="full")65 assert_array_almost_equal(z, r_z)66class TestConvolve(object):67 def test_object(self):68 d = [1.0] * 10069 k = [1.0] * 370 assert_array_almost_equal(convolve(d, k)[2:-2], cp.full(98, 3))71 def test_no_overwrite(self):72 d = cp.ones(100)73 k = cp.ones(3)74 convolve(d, k)75 assert_array_equal(d, cp.ones(100))...
test_utils.py
Source:test_utils.py
...19 raise AssertionError20 if not os.path.exists('./test_dir/models'):21 raise AssertionError22@pytest.mark.usefixtures('remove_file_structure')23def test_no_overwrite():24 if os.path.realpath(os.curdir).split("\\")[-1] != "test_dir":...
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!!