def test_adds_three_tables_to_old_table(self): tblA = Table(names=["colA", "colB"], data=[[0, 1], [0, 1]]) tblB = Table(names=["colA", "colB"], data=[[2, 3], [2, 3]]) tblC = Table(names=["colA", "colB"], data=[[4, 5], [4, 5]]) tblD = Table(names=["colA", "colB"], data=[[6, 7], [6, 7]]) tblE = rad_utils.combine_tables([tblB, tblC, tblD], tblA) assert np.all(tblE["colA"] == np.arange(8))
def test_prepend_table(self): tblA = Table(names=["colA", "colB"], data=[[0, 1], [0, 1]]) tblB = Table(names=["colA", "colB"], data=[[2, 3], [2, 3]]) tblC = rad_utils.combine_tables(tblB, tblA, prepend=True) assert np.all(tblC["colB"] == np.array([2, 3, 0, 1]))
def test_adds_3_tables_from_filename_to_nothing(self, input_tables): tblC = rad_utils.combine_tables(input_tables) assert len(tblC) == 19
def test_adds_table_from_filename_to_table_from_file(self, input_tables): tblA = input_tables[0] tblB = input_tables[1] tblC = rad_utils.combine_tables(tblB, tblA) assert len(tblC) == 6
def test_adds_table_from_filename_to_table_object(self, input_tables): tblA = ioascii.read(input_tables[0]) tblB = input_tables[1] tblC = rad_utils.combine_tables(tblB, tblA) assert len(tblC) == 6
def test_adds_table_from_filename_to_nothing(self, input_tables): tblA = ioascii.read(input_tables[0]) tblC = rad_utils.combine_tables(tblA) assert len(tblC) == 5
def test_adds_single_table(self): tblA = Table(names=["colA", "colB"], data=[[0, 1], [0, 1]]) tblC = rad_utils.combine_tables(tblA) assert np.all(tblC["colA"] == np.array([0, 1]))
def test_adds_two_tables(self): tblA = Table(names=["colA", "colB"], data=[[0, 1], [0, 1]]) tblB = Table(names=["colA", "colB"], data=[[2, 3], [2, 3]]) tblC = rad_utils.combine_tables(tblB, tblA) assert np.all(tblC["colB"] == np.array([0, 1, 2, 3]))