Esempio n. 1
0
    def test_filters(self):
        df = load(result_dir, a=2)
        self.assertEqual(df.shape[0], 2)
        self.assertEqual(df.a.sum(), 4)

        df = load(result_dir, a=[2])
        self.assertEqual(df.shape[0], 2)
        self.assertEqual(df.a.sum(), 4)

        df = load(result_dir, a=[1, 2])
        self.assertEqual(df.shape[0], 4)
        self.assertEqual(df.a.sum(), 6)

        df = load(result_dir, a=1000)
        self.assertTrue(df.empty)
Esempio n. 2
0
    def test_filters(self):
        df = load(result_dir, a=2)
        self.assertEqual(df.shape[0], 2)
        self.assertEqual(df.a.sum(), 4)

        df = load(result_dir, a=[2])
        self.assertEqual(df.shape[0], 2)
        self.assertEqual(df.a.sum(), 4)

        df = load(result_dir, a=[1,2])
        self.assertEqual(df.shape[0], 4)
        self.assertEqual(df.a.sum(), 6)

        df = load(result_dir, a=1000)
        self.assertTrue(df.empty)
Esempio n. 3
0
 def test_basics(self):
     df = load(result_dir)
     self.assertEqual(set(df.columns), set(['a', 'b', 'resfile', '_stream']))
     self.assertEqual(df.shape[0], 4)
Esempio n. 4
0
 def test_process(self):
     df = load(result_dir, a=1, _process=lambda x: pd.np.loadtxt(x).sum())
     self.assertAlmostEqual(df.sum()['res'], 2.7)
Esempio n. 5
0
 def test_stream(self):
     df = load(result_dir, _stream='stderr')
     self.assertTrue((df._stream == 'stderr').all())
Esempio n. 6
0
    def test_format(self):
        df = load(result_dir, b=0.3)
        self.assertTrue(df.empty)

        df = load(result_dir, b=0.3, _format={'b': '%.2f'})
        self.assertEqual(df.shape[0], 2)
Esempio n. 7
0
    def test_infer_types(self):
        df = load(result_dir)
        self.assertEqual(df.a.dtype, pd.np.int64)

        df = load(result_dir, _infer_types=False)
        self.assertEqual(df.a.dtype, pd.np.object_)
Esempio n. 8
0
 def test_basics(self):
     df = load(result_dir)
     self.assertEqual(set(df.columns), set(['a', 'b', 'resfile',
                                            '_stream']))
     self.assertEqual(df.shape[0], 4)
Esempio n. 9
0
 def test_process(self):
     df = load(result_dir, a=1, _process=lambda x: pd.np.loadtxt(x).sum())
     self.assertAlmostEqual(df.sum()['res'], 2.7)
Esempio n. 10
0
 def test_stream(self):
     df = load(result_dir, _stream='stderr')
     self.assertTrue((df._stream == 'stderr').all())
Esempio n. 11
0
    def test_format(self):
        df = load(result_dir, b=0.3)
        self.assertTrue(df.empty)

        df = load(result_dir, b=0.3, _format={'b': '%.2f'})
        self.assertEqual(df.shape[0], 2)
Esempio n. 12
0
    def test_infer_types(self):
        df = load(result_dir)
        self.assertEqual(df.a.dtype, pd.np.int64)

        df = load(result_dir, _infer_types=False)
        self.assertEqual(df.a.dtype, pd.np.object_)