def testTooFewColumnNames(self): """Test that too few names in config.colnames causes an error""" config = ReadTextCatalogTask.ConfigClass() for badColNames in ( ["name", "ra", "dec", "counts", "flux"], ["name"], ): config.colnames = badColNames config.header_lines = 1 task = ReadTextCatalogTask(config=config) with self.assertRaises(ValueError): task.run(TextPath)
def testGivenNames(self): """Test reading with column names in the config """ colnames = ("id", "ra_deg", "dec_deg", "total_counts", "total_flux", "is_resolved") config = ReadTextCatalogTask.ConfigClass() config.colnames = colnames config.header_lines = 1 task = ReadTextCatalogTask(config=config) arr = task.run(TextPath) self.assertEqual(arr.dtype.names, colnames) self.assertEqual(len(arr), 2) for inname, outname in zip(self.arr.dtype.names, colnames): self.assertTrue(np.array_equal(self.arr[inname], arr[outname]))