swego: add Copy method for CalcFlags

astrotools · Aug 6, 2016 · 5ed7f90 · 5ed7f90
1 parent b521a38
commit 5ed7f90
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/swego.go b/swego.go
@@ -37,6 +37,13 @@ type CalcFlags struct {
 	FileNameJPL string
 }
 
+// Copy returns a copy of the calculation flags fl.
+func (fl *CalcFlags) Copy() *CalcFlags {
+	copy := new(CalcFlags)
+	*copy = *fl
+	return copy
+}
+
 // Ephemeris represents an ephemeris implemented in the C library.
 type Ephemeris int32
 

diff --git a/swego_test.go b/swego_test.go
@@ -53,8 +53,18 @@ func TestNewHSys(t *testing.T) {
 	}
 }
 
+func TestCalcFlags_Copy(t *testing.T) {
+	fl := new(CalcFlags)
+	fl.Flags = FlagSpeed
+	got := fl.Copy()
+
+	if got == fl {
+		t.Errorf("%p = %p, want copy", got, fl)
+	}
+}
+
 func TestCalcFlags_SetEphemeris(t *testing.T) {
-	fl := CalcFlags{}
+	fl := new(CalcFlags)
 	fl.SetEphemeris(JPL)
 
 	if fl.Flags != int32(JPL) {