diff --git a/src/Deblan/Csv/Csv.php b/src/Deblan/Csv/Csv.php index db3c9d5..1054f20 100644 --- a/src/Deblan/Csv/Csv.php +++ b/src/Deblan/Csv/Csv.php @@ -63,6 +63,8 @@ class Csv { $this->setHasLegend(true); + $this->legend = $values; + $this->addLine($values, 0); } diff --git a/tests/CsvTest.php b/tests/CsvTest.php index 0120fff..4ae789c 100644 --- a/tests/CsvTest.php +++ b/tests/CsvTest.php @@ -17,7 +17,10 @@ class CsvTest extends \PHPUnit_Framework_TestCase $this->assertEquals(false, $csv->getHasLegend()); - $csv->setLegend(array('bim', 'bam')); + $csv->setLegend($a = array('bim', 'bam')); + + $this->assertEquals($a, $csv->getLegend()); + $csv->addLine(array('foo', 'bar')); $this->assertEquals(true, $csv->getHasLegend());