mirror of
https://github.com/hanzoai/numscript.git
synced 2026-10-30 18:26:57 +00:00
Alessandro ScandoneandGitHub
5dfc34cd3c
* feat: feature flag syntax * feat: change syntax * feat: reject invalid features * feat: add static check of unknown features * feat: more defensive args handling * fix: fix typo
146 lines
2.6 KiB
Go
146 lines
2.6 KiB
Go
package interpreter
|
|
|
|
import (
|
|
"math/big"
|
|
"testing"
|
|
|
|
"github.com/gkampitakis/go-snaps/snaps"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestFilterQuery(t *testing.T) {
|
|
fullBalance := Balances{
|
|
"alice": AccountBalance{
|
|
"EUR/2": big.NewInt(1),
|
|
"USD/2": big.NewInt(2),
|
|
},
|
|
"bob": AccountBalance{
|
|
"BTC": big.NewInt(3),
|
|
},
|
|
}
|
|
|
|
filteredQuery := fullBalance.filterQuery(BalanceQuery{
|
|
"alice": []string{"GBP/2", "YEN", "EUR/2"},
|
|
"bob": []string{"BTC"},
|
|
"charlie": []string{"ETH"},
|
|
})
|
|
|
|
require.Equal(t, BalanceQuery{
|
|
"alice": []string{"GBP/2", "YEN"},
|
|
"charlie": []string{"ETH"},
|
|
}, filteredQuery)
|
|
}
|
|
|
|
func TestCloneBalances(t *testing.T) {
|
|
fullBalance := Balances{
|
|
"alice": AccountBalance{
|
|
"EUR/2": big.NewInt(1),
|
|
"USD/2": big.NewInt(2),
|
|
},
|
|
"bob": AccountBalance{
|
|
"BTC": big.NewInt(3),
|
|
},
|
|
}
|
|
|
|
cloned := fullBalance.DeepClone()
|
|
|
|
fullBalance["alice"]["USD/2"].Set(big.NewInt(42))
|
|
|
|
require.Equal(t, big.NewInt(2), cloned["alice"]["USD/2"])
|
|
}
|
|
|
|
func TestPrettyPrintBalance(t *testing.T) {
|
|
fullBalance := Balances{
|
|
"alice": AccountBalance{
|
|
"EUR/2": big.NewInt(1),
|
|
"USD/1234": big.NewInt(999999),
|
|
},
|
|
"bob": AccountBalance{
|
|
"BTC": big.NewInt(3),
|
|
},
|
|
}
|
|
|
|
snaps.MatchSnapshot(t, fullBalance.PrettyPrint())
|
|
}
|
|
|
|
func TestCmpMaps(t *testing.T) {
|
|
|
|
b1 := Balances{
|
|
"alice": AccountBalance{
|
|
"EUR": big.NewInt(100),
|
|
},
|
|
}
|
|
|
|
b2 := Balances{
|
|
"alice": AccountBalance{
|
|
"EUR": big.NewInt(42),
|
|
},
|
|
}
|
|
|
|
require.Equal(t, false, CompareBalances(b1, b2))
|
|
}
|
|
|
|
func TestCmpMapsIncluding(t *testing.T) {
|
|
|
|
t.Run("including (subset)", func(t *testing.T) {
|
|
b2 := Balances{
|
|
"alice": AccountBalance{
|
|
"EUR": big.NewInt(100),
|
|
},
|
|
"bob": AccountBalance{
|
|
"EUR": big.NewInt(100),
|
|
},
|
|
}
|
|
|
|
b1 := Balances{
|
|
"alice": AccountBalance{
|
|
"EUR": big.NewInt(100),
|
|
},
|
|
}
|
|
|
|
require.Equal(t, true, CompareBalancesIncluding(b1, b2))
|
|
})
|
|
|
|
t.Run("different value", func(t *testing.T) {
|
|
b2 := Balances{
|
|
"alice": AccountBalance{
|
|
"EUR": big.NewInt(100),
|
|
},
|
|
"bob": AccountBalance{
|
|
"EUR": big.NewInt(100),
|
|
},
|
|
}
|
|
|
|
b1 := Balances{
|
|
"alice": AccountBalance{
|
|
"EUR": big.NewInt(0),
|
|
},
|
|
}
|
|
|
|
require.Equal(t, false, CompareBalancesIncluding(b1, b2))
|
|
})
|
|
|
|
t.Run("extra value", func(t *testing.T) {
|
|
b2 := Balances{
|
|
"alice": AccountBalance{
|
|
"EUR": big.NewInt(100),
|
|
},
|
|
"bob": AccountBalance{
|
|
"EUR": big.NewInt(100),
|
|
},
|
|
}
|
|
|
|
b1 := Balances{
|
|
"alice": AccountBalance{
|
|
"EUR": big.NewInt(100),
|
|
},
|
|
|
|
"extra-value": AccountBalance{
|
|
"EUR": big.NewInt(100),
|
|
},
|
|
}
|
|
|
|
require.Equal(t, false, CompareBalancesIncluding(b1, b2))
|
|
})
|
|
}
|