Python get_modules_and_shortcuts Examples

Example #1

 def test_no_extra_needed(self):
     command_list = ['df.reset_index()', 'df.head()']
     self.assertEqual(
         set(get_modules_and_shortcuts(command_list)),
         {
             ('pandas', 'pd'),
             ('dateutil', 'dateutil'),
         }
     )

Example #2

File: p_df_test.py Project: djangosporti/pandashells

 def test_get_extra_import_all_needed(self):
     command_list = [
         'pl.plot(df.x)', 'sns.distplot(df.x)', 'scp.stats.norm(1, 1)',
         'np.random.randn(1)'
     ]
     self.assertEqual(
         set(get_modules_and_shortcuts(command_list)),
         {
             ('dateutil', 'dateutil'),
             ('pandas', 'pd'),
             ('scipy', 'scp'),
             ('pylab', 'pl'),
             ('seaborn', 'sns'),
             ('numpy', 'np'),
         },
     )

Example #3

File: p_df_test.py Project: alayassir/pandashells

 def test_get_extra_import_all_needed(self):
     command_list = [
         'pl.plot(df.x)',
         'sns.distplot(df.x)',
         'scp.stats.norm(1, 1)',
         'np.random.randn(1)'
     ]
     self.assertEqual(
         set(get_modules_and_shortcuts(command_list)),
         {
             ('dateutil', 'dateutil'),
             ('pandas', 'pd'),
             ('scipy', 'scp'),
             ('pylab', 'pl'),
             ('seaborn', 'sns'),
             ('numpy', 'np'),
         },
     )