Beispiel #1
0
 def test_year_str(self):
     with self.assertRaises(TypeError):
         get_form_860(data_dir, year="2016")
Beispiel #2
0
 def test_good_year(self):
     form_860 = get_form_860(data_dir)
     self.assertIsInstance(form_860, pd.DataFrame)
Beispiel #3
0
 def test_bad_year(self):
     with self.assertRaises(ValueError):
         get_form_860(data_dir, year=3000)
Beispiel #4
0
 def test_bad_dir(self):
     bad_dir = path.abspath(path.join(path.dirname(__file__), "..", "foo"))
     with self.assertRaises(ValueError):
         get_form_860(bad_dir)